Use add_cleanup