fix import

Importing is broken when using on recent python. This fix will require 2.5+