Fix test-case