Slight edit to test cases