reward code fix

Signed-off-by: Andrew McKinney <mckinney@neuro.duke.edu>