Corrected missing import and syntax