small validation code change