Small validation change