Replace should_receive with expect...to receive