Fix stub in tests