Bugfix for calls via load_model.