test: Update edit test case