Change stubs in tests