Fixes test that was broken because the function definition changed.