Update the small test case