Test fix?

I think this is what is causing the test to fail.