Could you move this into `/tests/mock`?