Add a test for update correctness