Fix comment and import