Minor adjustment to the test cases