Fix bugs when import