Fix unused/broken import