Fix importing bug