diff --git a/UnitTest/ClassDiagramTests.cs b/UnitTest/ClassDiagramTests.cs
index 8674c798d577e62335d45b3056d74a7c91ecfde4..63d173c4f12b6e3afbdd0fe76afbb11d048fbfa6 100644
--- a/UnitTest/ClassDiagramTests.cs
+++ b/UnitTest/ClassDiagramTests.cs
@@ -206,7 +206,8 @@ public class ClassDiagramTests
 
             if (!isValid)
             {
-                Assert.Throws<ArgumentException>(() => ClassDiagramBuilder.BuildDiagram());
+                var xmi = ClassDiagramBuilder.BuildDiagram();
+                AssertCommonAssertions(xmi);
             }
             else
             {