small fix, models not defined