diff --git a/tests/drivers/test_tektronix_AWG5014C.py b/tests/drivers/test_tektronix_AWG5014C.py
index 7d01ff698151..f30d4f87ecf9 100644
--- a/tests/drivers/test_tektronix_AWG5014C.py
+++ b/tests/drivers/test_tektronix_AWG5014C.py
@@ -1,12 +1,12 @@
 import numpy as np
 import pytest
 
-from qcodes.instrument_drivers.tektronix.AWG5014 import Tektronix_AWG5014
+from qcodes.instrument_drivers.tektronix import TektronixAWG5014
 
 
 @pytest.fixture(scope="function")
 def awg():
-    awg_sim = Tektronix_AWG5014(
+    awg_sim = TektronixAWG5014(
         "awg_sim",
         address="GPIB0::1::INSTR",
         timeout=1,