diff --git a/force_bdss/tests/probe_classes.py b/force_bdss/tests/probe_classes.py
index b9b54da5486ea3d68f3f4c7692ca8cf51b94b221..54d7b6cbf01c3ba255fab5ea290bd419aea10f7c 100644
--- a/force_bdss/tests/probe_classes.py
+++ b/force_bdss/tests/probe_classes.py
@@ -41,6 +41,13 @@ class ProbeEvaluator(HasStrictTraits):
 
 
 class ProbeEvaluatorFactory(HasStrictTraits):
+    def __init__(self, plugin=None, *args, **kwargs):
+        if plugin is None:
+            plugin = mock.Mock(Plugin)
+
+        super(ProbeEvaluatorFactory, self).__init__(
+            plugin=plugin, *args, **kwargs)
+
     run_function = Function
 
     input_slots_type = Str('PRESSURE')
@@ -70,13 +77,6 @@ class ProbeKPICalculatorFactory(BaseKPICalculatorFactory,
     id = Str('enthought.test.datasource')
     name = Str('test_datasource')
 
-    def __init__(self, plugin=None, *args, **kwargs):
-        if plugin is None:
-            plugin = mock.Mock(Plugin)
-
-        super(ProbeKPICalculatorFactory, self).__init__(
-            plugin=plugin, *args, **kwargs)
-
     def create_model(self, model_data=None):
         return ProbeKPICalculatorModel(self, self.model_data)