Add test for update