Pass the model directly, yay simplification!