Fixed missing import for previous change