You should just add this as a `.cleanup`