diff --git a/DataTree.cc b/DataTree.cc
index db60f58b..f151fa3a 100644
--- a/DataTree.cc
+++ b/DataTree.cc
@@ -651,13 +651,15 @@ DataTree::writePowerDeriv(ostream &output, bool use_dll) const
            << "% The k-th derivative of x^p" << endl
            << "%" << endl
            << "function dxp=getPowerDeriv(x,p,k)" << endl
-           << "if (abs(x) < " << NEAR_ZERO << ") && (p > 0) && (k >= p) && (abs(p - round(p)) < " << NEAR_ZERO << ")" << endl
-           << "    dxp = 0;" << endl
-           << "else" << endl
-           << "    dxp = x^(p-k);" << endl
-           << "    for i=0:k-1" << endl
-           << "        dxp = dxp*p;" << endl
-           << "        p = p-1;" << endl
+           << "    if (abs(x) < " << NEAR_ZERO << ") && (p > 0) && (k >= p) && (abs(p - round(p)) < " << NEAR_ZERO << ")" << endl
+           << "        dxp = 0;" << endl
+           << "    else" << endl
+           << "        dxp = x^(p-k);" << endl
+           << "        for i=0:k-1" << endl
+           << "            dxp = dxp*p;" << endl
+           << "            p = p-1;" << endl
+           << "        end" << endl
            << "    end" << endl
            << "end" << endl;
+
 }
diff --git a/DynamicModel.cc b/DynamicModel.cc
index 07ff6e81..ee0673ec 100644
--- a/DynamicModel.cc
+++ b/DynamicModel.cc
@@ -795,6 +795,7 @@ DynamicModel::writeModelEquationsOrdered_M(const string &dynamic_basename) const
         default:
           break;
         }
+      output << "end" << endl;
       writePowerDeriv(output, false);
       output.close();
     }
@@ -2033,7 +2034,7 @@ DynamicModel::writeSparseDynamicMFile(const string &dynamic_basename, const stri
   open_par = false;
   mDynamicModelFile << "  oo_.endo_simul = y';\n";
   mDynamicModelFile << "return;\n";
-
+  mDynamicModelFile << "end" << endl;
   writePowerDeriv(mDynamicModelFile, false);
 
   mDynamicModelFile.close();
diff --git a/StaticModel.cc b/StaticModel.cc
index 73e77048..f072ff2b 100644
--- a/StaticModel.cc
+++ b/StaticModel.cc
@@ -407,6 +407,7 @@ StaticModel::writeModelEquationsOrdered_M(const string &static_basename) const
         default:
           break;
         }
+      output << "end" << endl;
       writePowerDeriv(output, false);
       output.close();
     }