Adjust unit test to test for update functionality.