diff --git a/test/py/u_boot_utils.py b/test/py/u_boot_utils.py
index 72d24e42aa33e684ede6c2ab9ac1834d1c58bdfa..9d243e047b19e1b1071b3036d6ad97a20b67b3a4 100644
--- a/test/py/u_boot_utils.py
+++ b/test/py/u_boot_utils.py
@@ -7,6 +7,7 @@
 import hashlib
 import os
 import os.path
+import pytest
 import sys
 import time