minor code improvememnt