Add test for after_update properly