Add test case to fix