diff --git a/gtsam/hybrid/tests/Switching.h b/gtsam/hybrid/tests/Switching.h
index 66dbc5fdc..49ced968e 100644
--- a/gtsam/hybrid/tests/Switching.h
+++ b/gtsam/hybrid/tests/Switching.h
@@ -214,13 +214,23 @@ struct Switching {
     return chain;
   }
 
-  HybridGaussianFactorGraph linearizedFactorGraph() {
+  /// Get the full linear factor graph.
+  HybridGaussianFactorGraph linearizedFactorGraph() const {
     HybridGaussianFactorGraph graph;
     graph.push_back(linearUnaryFactors);
     graph.push_back(linearBinaryFactors);
     graph.push_back(modeChain);
     return graph;
   }
+
+  /// Get all the nonlinear factors.
+  HybridNonlinearFactorGraph nonlinearFactorGraph() const {
+    HybridNonlinearFactorGraph graph;
+    graph.push_back(unaryFactors);
+    graph.push_back(binaryFactors);
+    graph.push_back(modeChain);
+    return graph;
+  }
 };
 
 }  // namespace gtsam