diff --git a/src/tests/common.py b/src/tests/common.py
index e87765a4..ece61454 100644
--- a/src/tests/common.py
+++ b/src/tests/common.py
@@ -13,7 +13,7 @@ _files = (
 def cleanup(home=None, files=_files):
     """Cleanup application files"""
     if not home:
-        import state
+        from pybitmessage import state
         home = state.appdata
     for pfile in files:
         try: