Small fix in validation