diff --git a/src/apps/molresponse/FrequencyResponse.cpp b/src/apps/molresponse/FrequencyResponse.cpp
index 97490425f7c..7742f189f5b 100644
--- a/src/apps/molresponse/FrequencyResponse.cpp
+++ b/src/apps/molresponse/FrequencyResponse.cpp
@@ -268,20 +268,6 @@ auto FrequencyResponse::update_response(World &world, X_space &chi, XCOperator<d
     if (iteration >= 0) {// & (iteration % 3 == 0)) {
         new_chi = kain_x_space_update(world, chi, new_res, kain_x_space);
     }
-    // here only accept the kain updates if residual_norm is bigger than max_rotaion
-    /*
-    for (const auto &b: chi.active) {
-        if (bsh_norms[b] > max_rotation) {
-            new_chi.x[b] = kain_chi.x[b];
-            if (r_params.calc_type() == "full") { new_chi.y[b] = kain_chi.y[b]; }
-        } else {
-            if (world.rank() == 0)
-                print("not accepting kain update since residual norm:", bsh_norms[b],
-                      " is smaller than max rotation: ", max_rotation);
-        }
-    }
-     */
-
 
     inner_to_json(world, "x_update", response_context.inner(new_chi, new_chi), iter_function_data);
 
diff --git a/src/apps/molresponse/ResponseBase.cpp b/src/apps/molresponse/ResponseBase.cpp
index b2e24ddb53b..01e17209d33 100644
--- a/src/apps/molresponse/ResponseBase.cpp
+++ b/src/apps/molresponse/ResponseBase.cpp
@@ -1195,12 +1195,16 @@ auto ResponseBase::kain_x_space_update(World &world, const X_space &chi, const X
 
         for (const auto &i: Chi.active) {
             auto temp = kain_x_space[i].update(x_vectors[i], x_residuals[i]);
+            truncate(world, temp);
             std::copy(temp.begin(), temp.begin() + n, kain_update.x[i].begin());
             std::copy(temp.begin() + n, temp.end(), kain_update.y[i].begin());
         };
     } else {
         // first compute the residuals
-        for (const auto &i: Chi.active) { kain_update.x[i] = kain_x_space[i].update(chi.x[i], residual_chi.x[i]); }
+        for (const auto &i: Chi.active) { kain_update.x[i] = kain_x_space[i].update(chi.x[i], residual_chi.x[i]);
+            // truncate the update
+            truncate(world, kain_update.x[i]);
+        }
     }
     if (r_params.print_level() >= 1) { molresponse::end_timer(world, "kain_x_update", "kain_x_update", iter_timing); }
     return kain_update;
@@ -1222,7 +1226,8 @@ void ResponseBase::x_space_step_restriction(World &world, const X_space &old_Chi
         auto m_new = to_response_matrix(temp);
         auto m_diff = to_response_matrix(diff);
 
-        for (size_t b = 0; b < m; b++) {
+        // only restrict active states
+        for (const auto &b: old_Chi.active) {
             auto step_size = norm2(world, m_diff[b]);
             auto norm_xb = norm2(world, m_old[b]);
             auto max_step = max_bsh_rotation;//norm;//* norm_xb;