Fix missing stub in tests