Fix check-model issue, a line of code was omitted during merge.