Attempt to fix import error