Also update the test to test for what it should

Yo dawg, I herd you liekd tests.