diff --git a/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py b/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py
index a42d39d6b..c33137d89 100644
--- a/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py
+++ b/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py
@@ -2,7 +2,7 @@
     from qcodes.tests.driver_test_case import DriverTestCase
 except ImportError:
     # qcodes < 0.37
-    from qcodes.instrument_drivers.test import DriverTestCase  # type: ignore[import,no-redef]
+    from qcodes.instrument_drivers.test import DriverTestCase  # type: ignore[import-not-found,no-redef]
 import unittest
 
 try: