fixed an import and a comment