Fix more of the post update bug.