Fix missing import and fix typo