Fix page refresh issues on draft completion