diff --git a/inference/compute/pt/pytorch_linear.py b/inference/compute/pt/pytorch_linear.py
index 9c946be9..7fea918d 100644
--- a/inference/compute/pt/pytorch_linear.py
+++ b/inference/compute/pt/pytorch_linear.py
@@ -33,7 +33,7 @@ def forward(self, x):
 
 
 def infer_nnpi(model, device, data_type, input_size, output_size, batch_size, args):
-    import torch_glow
+    import glow.fb.torch_glow.py.NNPI.torch_glow as torch_glow
 
     # Detailed structure for spec can be found at https://fburl.com/diffusion/79q4efud
     # Create compilation spec