diff --git a/GPy/testing/grid_tests.py b/GPy/testing/grid_tests.py
index c6aaf049..7739f233 100644
--- a/GPy/testing/grid_tests.py
+++ b/GPy/testing/grid_tests.py
@@ -3,13 +3,12 @@
 
 # Kurt Cutajar
 
-import unittest
 import numpy as np
 import GPy
 
 
-class GridModelTest(unittest.TestCase):
-    def setUp(self):
+class GridModelTest:
+    def setup(self):
         ######################################
         # # 3 dimensional example