Fixed a broken import