Correct missing import