Fixed another missing import