Tiny clean up