diff --git a/src/test/java/de/hhu/stups/codegenerator/java/TestMachines.java b/src/test/java/de/hhu/stups/codegenerator/java/TestMachines.java
index 4ebd883c9fe80577c08bb7fc2b450d4e3f126171..d7f4d330f76ecded570112cc5c48bcdd84ec1c36 100644
--- a/src/test/java/de/hhu/stups/codegenerator/java/TestMachines.java
+++ b/src/test/java/de/hhu/stups/codegenerator/java/TestMachines.java
@@ -1,7 +1,7 @@
 package de.hhu.stups.codegenerator.java;
 
 import de.hhu.stups.codegenerator.generators.CodeGenerationException;
-import de.prob.parser.ast.visitors.TypeErrorException;
+
 import org.junit.Ignore;
 import org.junit.Test;
 
@@ -768,11 +768,7 @@ public class TestMachines extends TestJava {
 
     @Test
     public void testDroneMainController() throws Exception {
-        try {
-            testJava("drone/DroneMainController");
-        } catch (TypeErrorException e) {
-            System.out.println(e.getMessage());
-        }
+        testJava("drone/DroneMainController");
     }
 
 }
\ No newline at end of file
diff --git a/src/test/resources/de/hhu/stups/codegenerator/drone/DroneMainController.mch b/src/test/resources/de/hhu/stups/codegenerator/drone/DroneMainController.mch
index 11d756c2750f8359328b9f861f06c0355316c4c5..7733c48bb01205a5926b77a80a3726c6eec4d76e 100644
--- a/src/test/resources/de/hhu/stups/codegenerator/drone/DroneMainController.mch
+++ b/src/test/resources/de/hhu/stups/codegenerator/drone/DroneMainController.mch
@@ -1,5 +1,5 @@
 MACHINE DroneMainController
-INCLUDES DroneSafetyController, DroneCommunicator
+INCLUDES DroneSafetyController, DroneCommunicator_Mockup
 
 OPERATIONS