Fixed missing import statement