Fixed broken test case