fix reward when only one task