Reward is now correctly a float, and not a list containing a string.