Cleanup after update