diff --git a/torch_glow/torch_glow/to_glow.py b/torch_glow/torch_glow/to_glow.py
index b8516b847..deb41b435 100644
--- a/torch_glow/torch_glow/to_glow.py
+++ b/torch_glow/torch_glow/to_glow.py
@@ -6,8 +6,9 @@
 from enum import Enum
 from typing import Any, Iterable, List, Mapping, Optional, Tuple, Union
 
+import glow.fb.torch_glow.py.FBA.torch_glow as torch_glow
+
 import torch
-import torch_glow
 
 __all__ = [
     "to_glow",