fix cleanup script