diff --git a/test/py/u_boot_utils.py b/test/py/u_boot_utils.py
index 72d24e42aa..9d243e047b 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