diff --git a/test/py/tests/test_extension.py b/test/py/tests/test_extension.py
index d0840f7..6122349 100644
--- a/test/py/tests/test_extension.py
+++ b/test/py/tests/test_extension.py
@@ -6,7 +6,7 @@
 
 import os
 import pytest
-import u_boot_utils
+import utils
 
 overlay_addr = 0x1000