diff --git a/zetta_utils/tensor_ops/__init__.py b/zetta_utils/tensor_ops/__init__.py
index 9b05678e3..44d1880ec 100644
--- a/zetta_utils/tensor_ops/__init__.py
+++ b/zetta_utils/tensor_ops/__init__.py
@@ -1,6 +1,3 @@
-from . import generators, traceback_supress
-
-from . import common, convert, label, mask, multitensor, normalization
 from .common import (
     InterpolationMode,
     compare,
@@ -15,3 +12,8 @@
 from .convert import astype, to_np, to_torch
 from .label import get_disp_pair, seg_to_aff, seg_to_rgb
 from .mask import filter_cc  # , coarsen
+
+# Circular import otherwise
+from . import generators, traceback_supress
+
+from . import common, convert, label, mask, multitensor, normalization