fix the import

quick fix for module import