diff --git a/tests/test_autograd.py b/tests/test_autograd.py
index d70cddd..a042ce6 100644
--- a/tests/test_autograd.py
+++ b/tests/test_autograd.py
@@ -1,6 +1,6 @@
 import unittest
 import norch
-from norch import utils
+from norch.utils import utils_unittests as utils
 import torch
 import os
 
diff --git a/tests/test_operations.py b/tests/test_operations.py
index ed72ec6..e3fac6a 100644
--- a/tests/test_operations.py
+++ b/tests/test_operations.py
@@ -1,6 +1,6 @@
 import unittest
 import norch
-from norch import utils
+from norch.utils import utils_unittests as utils
 import torch
 import sys
 import os