fix comment logic