Fix StubGen test