Fix cleanup script