Tiny code improvement