Fixed @cvan's code

Someone didn't run their tests!