diff --git a/tests/bm_cubify.py b/tests/bm_cubify.py
index 041dabfea..37bd77422 100644
--- a/tests/bm_cubify.py
+++ b/tests/bm_cubify.py
@@ -4,7 +4,8 @@
 from fvcore.common.benchmark import benchmark
 
 from test_cubify import TestCubify
-
+import os 
+import time
 
 def bm_cubify() -> None:
     kwargs_list = [