fixing broken import on comments