diff --git a/src/poly/utils/private/operator.cpp b/src/poly/utils/private/operator.cpp
index 9951f14..04fc048 100644
--- a/src/poly/utils/private/operator.cpp
+++ b/src/poly/utils/private/operator.cpp
@@ -267,9 +267,10 @@ mfem::Operator& PolytropeOperator::GetGradient(const mfem::Vector &x) const {
     // auto *gradPtr = &grad;
     // updatePreconditioner(grad);
 
-    // // TODO: Save the state vector somewhere so that I can inspect how the actual solution
-    // // to theta and phi are evolving
-    // m_jacobian->SetBlock(0, 0, &grad);
+    // TODO: Save the state vector somewhere so that I can inspect how the actual solution
+    // to theta and phi are evolving
+    m_jacobian->SetBlock(0, 0, &grad);
+
     // // TODO: Save the jacobian here
     // std::cout << "Writing out jacobian to file..." << std::endl;
     // std::vector<mfem::Operator*> ops;
@@ -280,6 +281,7 @@ mfem::Operator& PolytropeOperator::GetGradient(const mfem::Vector &x) const {
     // saveBlockFormToBinary(ops, {{0, 0}, {0, 1}, {1, 0}, {1, 1}}, {false, false, false, false}, "jacobian_" + std::to_string(s_newtonStepGrad) + ".bin");
     // s_newtonStepGrad++;
     // std::cout << "Done writing out jacobian to file." << std::endl;
+
     return *m_jacobian;
 }