Add a small cleanup tweak