Fix validation error

Fix validation