diff --git a/torch_glow/tests/backends/NNPI/model_serialization_test.py b/torch_glow/tests/backends/NNPI/model_serialization_test.py
index fe4b3079c..26950f378 100644
--- a/torch_glow/tests/backends/NNPI/model_serialization_test.py
+++ b/torch_glow/tests/backends/NNPI/model_serialization_test.py
@@ -3,9 +3,10 @@
 
 import unittest
 
+import glow.fb.torch_glow.py.NNPI.torch_glow as torch_glow
+
 import torch
 import torch.jit
-import torch_glow
 from glow.glow.torch_glow.tests.tests import utils