tiny code improvement