Update model check

use simple check for model choice