diff --git a/src/main/java/teetime/framework/test/StageTester.java b/src/main/java/teetime/framework/test/StageTester.java
index 550ccfcf84e0fbd47dac673bb94c3d4aaca8ca97..e24062d672cfaecb42a30a82c4039a5e18f855d0 100644
--- a/src/main/java/teetime/framework/test/StageTester.java
+++ b/src/main/java/teetime/framework/test/StageTester.java
@@ -18,8 +18,6 @@ package teetime.framework.test;
 import java.util.ArrayList;
 import java.util.List;
 
-import junit.framework.AssertionFailedError;
-
 import teetime.framework.Analysis;
 import teetime.framework.AnalysisConfiguration;
 import teetime.framework.InputPort;
@@ -85,7 +83,7 @@ public final class StageTester {
 		@SuppressWarnings("unchecked")
 		public StageTester to(final InputPort<I> port) {
 			if (port.getOwningStage() != stage) {
-				throw new AssertionFailedError();
+				throw new AssertionError();
 			}
 			this.port = (InputPort<Object>) port;