Add test for .update