fix the import bug