diff --git a/rtos/tests/TESTS/host_tests/timing_drift_auto.py b/rtos/tests/TESTS/host_tests/timing_drift_auto.py
index 65ac90a160..912cb3452b 100644
--- a/rtos/tests/TESTS/host_tests/timing_drift_auto.py
+++ b/rtos/tests/TESTS/host_tests/timing_drift_auto.py
@@ -16,7 +16,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from mbed_host_tests import BaseHostTest
+from htrun import BaseHostTest
 import time