Move update from before to after_update