fix re-run bug