diff --git a/debug-tests/debug_utils.py b/debug-tests/debug_utils.py
index a7029a78b..43a11fc31 100644
--- a/debug-tests/debug_utils.py
+++ b/debug-tests/debug_utils.py
@@ -6,6 +6,8 @@
 import cupy
 import rmm
 
+from .utils import get_num_gpus
+
 ITERATIONS = 100