diff --git a/atorch/atorch/distributed/distributed.py b/atorch/atorch/distributed/distributed.py
index 5e0893175..91e521d13 100644
--- a/atorch/atorch/distributed/distributed.py
+++ b/atorch/atorch/distributed/distributed.py
@@ -608,8 +608,7 @@ def init_distributed(
                 return False
         elif backend == "accl":
             try:
-                # noqa: F401
-                import torch_accl
+                import torch_accl # noqa: F401
             except ImportError:
                 logger.error("import torch_accl failed")
                 return False