Clean up after update