add test case back