diff --git a/tests/code/MultiFormTestClass.php b/tests/code/MultiFormTestClass.php
index b0d1e56..220295f 100644
--- a/tests/code/MultiFormTestClass.php
+++ b/tests/code/MultiFormTestClass.php
@@ -1,6 +1,6 @@
 <?php
 
-class MultiFormTestClass extends MultiForm {
+class MultiFormTestClass extends MultiForm implements TestOnly {
 	
 	protected static $start_step = 'MultiFormTestStepOne';
 	
diff --git a/tests/code/MultiFormTestStepOne.php b/tests/code/MultiFormTestStepOne.php
index 3da22ae..aa7463e 100644
--- a/tests/code/MultiFormTestStepOne.php
+++ b/tests/code/MultiFormTestStepOne.php
@@ -1,6 +1,6 @@
 <?php
 
-class MultiFormTestStepOne extends MultiFormStep {
+class MultiFormTestStepOne extends MultiFormStep implements TestOnly {
 	
 	protected static $next_steps = 'MultiFormTestStepTwo';
 	
diff --git a/tests/code/MultiFormTestStepThree.php b/tests/code/MultiFormTestStepThree.php
index 7576ac9..ef659e2 100644
--- a/tests/code/MultiFormTestStepThree.php
+++ b/tests/code/MultiFormTestStepThree.php
@@ -1,6 +1,6 @@
 <?php
 
-class MultiFormTestStepThree extends MultiFormStep {
+class MultiFormTestStepThree extends MultiFormStep implements TestOnly {
 	
 	protected static $is_final_step = true;
 	
diff --git a/tests/code/MultiFormTestStepTwo.php b/tests/code/MultiFormTestStepTwo.php
index 4d7f901..d4a3366 100644
--- a/tests/code/MultiFormTestStepTwo.php
+++ b/tests/code/MultiFormTestStepTwo.php
@@ -1,6 +1,6 @@
 <?php
 
-class MultiFormTestStepTwo extends MultiFormStep {
+class MultiFormTestStepTwo extends MultiFormStep implements TestOnly {
 	
 	protected static $next_steps = 'MultiFormTestStepThree';