minor change on the test case