Add test case to fix issue