Fix another broken import