Fix errornous import