Just a small cleanup