diff --git a/VERSION_NUMBER b/VERSION_NUMBER
index ec96a6218..e5a9958c3 100644
--- a/VERSION_NUMBER
+++ b/VERSION_NUMBER
@@ -1 +1 @@
-0.3.16
+0.3.17
diff --git a/onnxoptimizer/optimize.h b/onnxoptimizer/optimize.h
index f25e6c3f6..9404b8761 100644
--- a/onnxoptimizer/optimize.h
+++ b/onnxoptimizer/optimize.h
@@ -46,12 +46,23 @@ struct Optimizer {
     ModelProto mp_out = PrepareOutput(mp_in);
     this->pass_manager->run(*g);
     ExportModelProto(&mp_out, g);
+
+    // Maybe we can optimize these functions, now just copy
+    AddFunctionsToModel(mp_in, mp_out);
     return mp_out;
   }
 
  private:
   std::shared_ptr<PassManager> pass_manager;
 
+  void AddFunctionsToModel(const ModelProto &original_model,
+                           ModelProto &output_model) {
+    for (const auto& function_proto : original_model.functions()) {
+      auto* p_f = output_model.add_functions();
+      p_f->CopyFrom(function_proto);
+    }
+  }
+
   ModelProto AddInitializerToInput(const ModelProto &original_model) {
     ModelProto model = original_model;
     std::vector<std::string> input_names;