Fixed missing import, fixed typo.