diff --git a/py_gen/tests/of10.py b/py_gen/tests/of10.py
index 597de5f..86f9b2b 100644
--- a/py_gen/tests/of10.py
+++ b/py_gen/tests/of10.py
@@ -44,6 +44,7 @@
         self.assertTrue(hasattr(ofp, "message"))
 
     def test_version(self):
+        import loxi
         self.assertTrue(hasattr(loxi.of10, "ProtocolError"))
         self.assertTrue(hasattr(loxi.of10, "OFP_VERSION"))
         self.assertEquals(loxi.of10.OFP_VERSION, 1)