Remove extra import and fix comment