fix import statement bug