Adding mock to specs