Change mock generation for tests

No match.