Test case fix again