hack to pass back the model