Update to get proper reward reduction