fix missing model assignment