fixed the bug when import