Add test for updates