Having a hard time figuring out how to test this change. I think this is the right test, but I can't figure out how the stubs work here.