diff --git a/main.cpp b/main.cpp
index a86fd01..a601263 100644
--- a/main.cpp
+++ b/main.cpp
@@ -105,7 +105,7 @@ vector<MyFunction> get_function_class(MyFunction f, const vector< MyMatrix >& tr
         cur_res.insert(cur_f);
     }
 
-    set<MyFunction> transformed_res;
+    set<MyFunction> transformed_res(cur_res.begin(), cur_res.end());
     for (auto f: cur_res) {
         for (auto transformation: tranformations) {
             MyFunction linear_transformed = get_function_from_callable<Storage, ARGS_COUNT>(