diff --git a/com.archimatetool.script.tests/src/com/archimatetool/script/dom/model/DiagramModelReferenceProxyTests.java b/com.archimatetool.script.tests/src/com/archimatetool/script/dom/model/DiagramModelReferenceProxyTests.java
index c2709aa9..428e304f 100644
--- a/com.archimatetool.script.tests/src/com/archimatetool/script/dom/model/DiagramModelReferenceProxyTests.java
+++ b/com.archimatetool.script.tests/src/com/archimatetool/script/dom/model/DiagramModelReferenceProxyTests.java
@@ -37,6 +37,19 @@ public void runOnceBeforeEachTest() {
         actualTestProxy = (DiagramModelReferenceProxy)testProxy;
     }
 
+    @Override
+    @Test
+    public void getType() {
+        // The type should be the IDiagramModelReference, not the referenced diagram model
+        assertEquals(ModelUtil.getKebabCase(testProxy.getEObject().eClass().getName()), testProxy.getType());
+    }
+    
+    @Override
+    public void attr_Type() {
+        // The type should be the IDiagramModelReference, not the referenced diagram model        
+        assertEquals(ModelUtil.getKebabCase(testProxy.getEObject().eClass().getName()), testProxy.getType());
+    }
+    
     @Override
     @Test
     public void get_ReturnsCorrectProxy() {
diff --git a/com.archimatetool.script/src/com/archimatetool/script/dom/model/DiagramModelReferenceProxy.java b/com.archimatetool.script/src/com/archimatetool/script/dom/model/DiagramModelReferenceProxy.java
index e599037d..4f88766c 100644
--- a/com.archimatetool.script/src/com/archimatetool/script/dom/model/DiagramModelReferenceProxy.java
+++ b/com.archimatetool.script/src/com/archimatetool/script/dom/model/DiagramModelReferenceProxy.java
@@ -34,4 +34,9 @@ protected EObject getReferencedEObject() {
         return getEObject().getReferencedModel();
     }
 
+    @Override
+    public String getType() {
+        // The type should be the IDiagramModelReference, not the referenced diagram model
+        return ModelUtil.getKebabCase(getEObject().eClass().getName());
+    }
 }