From 1b17e9a693be64e9182df0f7168874630ba6693f Mon Sep 17 00:00:00 2001
From: hbruintjes <h.bruintjes@cs.rwth-aachen.de>
Date: Thu, 30 Jun 2016 17:15:32 +0200
Subject: [PATCH] Update core, builder returns model directly

Former-commit-id: a30dffa37d5cdbdd05887cc031ba00664dd86add
---
 stormpy/src/core/core.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/stormpy/src/core/core.cpp b/stormpy/src/core/core.cpp
index 73f95873b..b8f696890 100644
--- a/stormpy/src/core/core.cpp
+++ b/stormpy/src/core/core.cpp
@@ -31,7 +31,7 @@ void define_parse(py::module& m) {
 // Thin wrapper for model building
 template<typename ValueType>
 std::shared_ptr<storm::models::ModelBase> buildModel(storm::prism::Program const& program, std::shared_ptr<storm::logic::Formula const> const& formula) {
-    return storm::buildSymbolicModel<ValueType>(program, std::vector<std::shared_ptr<storm::logic::Formula const>>(1,formula)).model;
+    return storm::buildSymbolicModel<ValueType>(program, std::vector<std::shared_ptr<storm::logic::Formula const>>(1,formula));
 }
 
 void define_build(py::module& m) {
@@ -40,4 +40,5 @@ void define_build(py::module& m) {
     m.def("_build_parametric_model", &buildModel<storm::RationalFunction>, "Build the parametric model", py::arg("program"), py::arg("formula"));
     m.def("build_model_from_prism_program", &storm::buildSymbolicModel<double>, "Build the model", py::arg("program"), py::arg("formulas"));
     m.def("build_parametric_model_from_prism_program", &storm::buildSymbolicModel<storm::RationalFunction>, "Build the parametric model", py::arg("program"), py::arg("formulas"));
+    
 }