diff --git a/ttool/src/test/java/tmltranslator/dsez3engine/InputInstanceTest.java b/ttool/src/test/java/tmltranslator/dsez3engine/InputInstanceTest.java
index 0dc2ccf6a8f6f58487fae6eaea92b02e0bc47b1f..c3e95ab994a68e8892533a386ade512ba30918de 100644
--- a/ttool/src/test/java/tmltranslator/dsez3engine/InputInstanceTest.java
+++ b/ttool/src/test/java/tmltranslator/dsez3engine/InputInstanceTest.java
@@ -321,10 +321,10 @@ public class InputInstanceTest {
             tempTasks.add((TMLTask) inputInstance.getModeling().getTasks().get(i));
         }
 
-        assertEquals(inputInstance.getBufferIn((TMLTask) tempTasks.get(0)), 0);
-        assertEquals(inputInstance.getBufferIn((TMLTask) tempTasks.get(1)), 2);
-        assertEquals(inputInstance.getBufferIn((TMLTask) tempTasks.get(2)), 2);
-        assertEquals(inputInstance.getBufferIn((TMLTask) tempTasks.get(3)), 10);
+        assertEquals(inputInstance.getBufferIn( tempTasks.get(0)), 0);
+        assertEquals(inputInstance.getBufferIn( tempTasks.get(1)), 2);
+        assertEquals(inputInstance.getBufferIn( tempTasks.get(2)), 2);
+        assertEquals(inputInstance.getBufferIn(tempTasks.get(3)), 10);
 
 
     }
@@ -336,10 +336,10 @@ public class InputInstanceTest {
             tempTasks.add((TMLTask) inputInstance.getModeling().getTasks().get(i));
         }
 
-        assertEquals(inputInstance.getBufferOut((TMLTask) tempTasks.get(0)), 4);
-        assertEquals(inputInstance.getBufferOut((TMLTask) tempTasks.get(1)), 5);
-        assertEquals(inputInstance.getBufferOut((TMLTask) tempTasks.get(2)), 5);
-        assertEquals(inputInstance.getBufferOut((TMLTask) tempTasks.get(3)), 0);
+        assertEquals(inputInstance.getBufferOut( tempTasks.get(0)), 4);
+        assertEquals(inputInstance.getBufferOut( tempTasks.get(1)), 5);
+        assertEquals(inputInstance.getBufferOut( tempTasks.get(2)), 5);
+        assertEquals(inputInstance.getBufferOut( tempTasks.get(3)), 0);
 
     }