please update your test case as well