Add post_cleanup_hook