minor change to the model picking demo