Fix issue in cleanup script