Fix broken import