Fixing re-run button