Removing duplicate arg-max-reached check