Fix validation error

Fix validation error