Fix missing import and typo