Add a test for the update mode