tiny fix
Fix comments