Fix broken import path