Fix: remaining old faulty import