Fix clean up script