Fixed an import bug