fix importing bug