Fixes test that I accidentally broke because I'm a naughty boy