small fix in trial model