Could we move the mock to a separate file?