Fixing import error and typo.