fix import error running setup