diff --git a/test/py/tests/test_net.py b/test/py/tests/test_net.py
index 52ecf93..9f4ee0c 100644
--- a/test/py/tests/test_net.py
+++ b/test/py/tests/test_net.py
@@ -5,7 +5,7 @@
 # tftpboot commands.
 
 import pytest
-import u_boot_utils
+import utils
 import uuid
 import datetime
 import re
@@ -315,7 +315,7 @@
 
     addr = f.get('addr', None)
     if not addr:
-        addr = u_boot_utils.find_ram_base(ubman)
+        addr = utils.find_ram_base(ubman)
 
     fn = f['fn']
     output = ubman.run_command('nfs %x %s' % (addr, fn))
@@ -411,7 +411,7 @@
 
     addr = f.get("addr", None)
     if not addr:
-        addr = u_boot_utils.find_ram_base(ubman)
+        addr = utils.find_ram_base(ubman)
 
     sz = f.get("size", None)
     timeout = f.get("timeout", ubman.p.timeout)