fix import bug

fixed import bug