Fix leftover import