diff --git a/wrapper.py b/wrapper.py
index 999c1b6..b8e8bd3 100644
--- a/wrapper.py
+++ b/wrapper.py
@@ -1,10 +1,10 @@
 # coding=utf-8
 
-from module_state import ModuleState
+from xmlparser.node import ModuleState
 
 class Wrapper:
   """Simulate a hash table
-  
+
   """
 
   def __init__(self):