Fix comments slightly