diff --git a/test/spec/features/modeling/AppendShapeSpec.js b/test/spec/features/modeling/AppendShapeSpec.js
index 5ad0498f..5487dd08 100644
--- a/test/spec/features/modeling/AppendShapeSpec.js
+++ b/test/spec/features/modeling/AppendShapeSpec.js
@@ -143,6 +143,8 @@ describe('features/modeling - append shape', function() {
           }),
           connectionDi = getDi(elementRegistry.get(connection.id));
 
+      // assume
+      expect(connectionDi).to.exist;
 
       // when
       commandStack.undo();
@@ -176,6 +178,9 @@ describe('features/modeling - append shape', function() {
           }),
           connectionDi = getDi(elementRegistry.get(connection.id));
 
+      // assume
+      expect(connectionDi).to.exist;
+
       // when
       commandStack.undo();
 
@@ -211,6 +216,9 @@ describe('features/modeling - append shape', function() {
           }),
           connectionDi = getDi(elementRegistry.get(connection.id));
 
+      // assume
+      expect(connectionDi).to.exist;
+
       // when
       commandStack.undo();
 
@@ -272,6 +280,9 @@ describe('features/modeling - append shape', function() {
           }),
           connectionDi = getDi(elementRegistry.get(connection.id));
 
+      // assume
+      expect(connectionDi).to.exist;
+
       // when
       commandStack.undo();
       commandStack.redo();