Fixed a broekn testcase