diff --git a/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py b/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py
index e7170938a..2c4f372ec 100644
--- a/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py
+++ b/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py
@@ -1,4 +1,8 @@
-from qcodes.instrument_drivers.test import DriverTestCase
+try:
+    from qcodes.tests.driver_test_case import DriverTestCase
+except ImportError:
+    # qcodes < 0.37
+    from qcodes.instrument_drivers.test import DriverTestCase
 import unittest
 
 try: