Add test from two test case for persistance