Add custom mocks to ui specs