Fixed the testcase so it actually works