Update specs to enable mocking