Fix another broken imports