Fix broken stub in tests.