Fixed a bug in which the wrong reward was being observed. Closes #1