Fixed a test case issue