Fix broken test.

I am an idiot, and rushed my commit this morning. The tests are now fixed.