Add do_post_update support