Fix missing validation