Switched to Mockery for object mocking