Add a test for the update callback