diff --git a/src/FDM/YASim/Model.cpp b/src/FDM/YASim/Model.cpp
index 4ca6d95da..a5d442fe4 100644
--- a/src/FDM/YASim/Model.cpp
+++ b/src/FDM/YASim/Model.cpp
@@ -168,7 +168,7 @@ void Model::iterate()
 void Model::setState(State* s)
 {
     _integrator.setState(s);
-    _s = s;
+    _s = _integrator.getState();
 }