Correcting validation issue