Fix typo and broken import