1 files changed, 4 insertions, 1 deletions
diff --git a/test/orm/compile.py b/test/orm/compile.py
index 61107ce8e..23f04db85 100644
--- a/test/orm/compile.py
+++ b/test/orm/compile.py
@@ -1,7 +1,10 @@
 import testbase
 from sqlalchemy import *
+from sqlalchemy.orm import *
+from testlib import *
 
-class CompileTest(testbase.AssertMixin):
+
+class CompileTest(AssertMixin):
     """test various mapper compilation scenarios"""
     def tearDownAll(self):
         clear_mappers()