diff --git a/src/python/oftest/illegal_message.py b/src/python/oftest/illegal_message.py
index f225b90..8e52e33 100644
--- a/src/python/oftest/illegal_message.py
+++ b/src/python/oftest/illegal_message.py
@@ -3,10 +3,10 @@
 """
 
 import struct
-import of10
+import ofp
 
 class illegal_message_type(object):
-    version = of10.OFP_VERSION
+    version = ofp.OFP_VERSION
     type = 217
 
     def __init__(self, xid=None):