It feels odd to have test code create logical branches in our production code, would we be able to update the failing specs to setup the mocks correctly or avoid the mocks entirely?