Correct broken import