1 files changed, 4 insertions, 6 deletions
diff --git a/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testinterfaces.py b/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testinterfaces.py
index 9a064663e..ace509057 100644
--- a/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testinterfaces.py
+++ b/testsuite/Testsrc/Testlib/TestServer/TestPlugin/Testinterfaces.py
@@ -326,13 +326,11 @@ class TestGoalValidator(Bcfg2TestCase):
                           gv.validate_goals, None, None)
 
 
-class TestVersion(TestPlugin):
+class TestVersion(Bcfg2TestCase):
     test_obj = Version
-
-    def get_obj(self, core=None):
-        if core is None:
-            core = Mock()
-        return self.test_obj(core, datastore)
+    
+    def get_obj(self):
+        return self.test_obj(datastore)
 
     def test_get_revision(self):
         d = self.get_obj()