Fix removed import