Fix another import