Added a tiny fix to make the testcase work again.