Fixed Reinforcement logic and tests