diff --git a/src/feature_creation/feature_space/FeatureSpace.cpp b/src/feature_creation/feature_space/FeatureSpace.cpp
index 086a1cfa6926e202e30c84890d3847c6fb9a3209..4c5419f3c570ad412ab30c376639d4fa40ba102d 100644
--- a/src/feature_creation/feature_space/FeatureSpace.cpp
+++ b/src/feature_creation/feature_space/FeatureSpace.cpp
@@ -689,7 +689,7 @@ void FeatureSpace::project_generated(const double* prop, const int size, std::ve
     {
         std::vector<node_ptr> phi_sel_private(phi_sel);
         std::vector<double> scores_sel_private(scores_sel);
-        int index_base = node_value_arrs::N_STORE_FEATURES + _n_sis_select * (omp_get_thread_num() + _mpi_comm->size());
+        int index_base = _phi.size() + _n_sis_select * (omp_get_thread_num() + _mpi_comm->size());
 
         #ifdef PARAMETERIZE
         std::shared_ptr<NLOptimizer> optimizer = nlopt_wrapper::get_optimizer(_project_type, _task_sizes, _prop, _max_phi, _max_param_depth);
@@ -698,7 +698,7 @@ void FeatureSpace::project_generated(const double* prop, const int size, std::ve
         #pragma omp for schedule(dynamic)
         for(auto feat = _phi.begin() + _start_gen.back() + _mpi_comm->rank(); feat < _phi.end(); feat += _mpi_comm->size())
         {
-            unsigned long int feat_ind = node_value_arrs::N_STORE_FEATURES + _n_sis_select * (omp_get_num_threads() + _mpi_comm->size());
+            unsigned long int feat_ind = _phi.size() + _n_sis_select * (omp_get_num_threads() + _mpi_comm->size());
 
             node_value_arrs::clear_temp_reg_thread();
             std::vector<node_ptr> generated_phi;
@@ -774,7 +774,7 @@ void FeatureSpace::project_generated(const double* prop, const int size, std::ve
 
         #pragma omp critical
         {
-            index_base = node_value_arrs::N_STORE_FEATURES + _n_sis_select * _mpi_comm->rank();
+            index_base = _phi.size() + _n_sis_select * _mpi_comm->rank();
             worst_score_ind = std::max_element(scores_sel.begin(), scores_sel.end()) - scores_sel.begin();
             for(int sc = 0; sc < scores_sel_private.size(); ++sc)
             {