diff --git a/python/gtsam/examples/Pose2SLAMExample_lago.py b/python/gtsam/examples/Pose2SLAMExample_lago.py
index b50b38759..d8cddde0b 100644
--- a/python/gtsam/examples/Pose2SLAMExample_lago.py
+++ b/python/gtsam/examples/Pose2SLAMExample_lago.py
@@ -25,12 +25,7 @@ from argparse import Namespace
 import numpy as np
 
 import gtsam
-from gtsam import Pose2, PriorFactorPose2, Values
-
-
-def vector3(x: float, y: float, z: float) -> np.ndarray:
-    """Create 3d double numpy array."""
-    return np.array([x, y, z], dtype=float)
+from gtsam import Point3, Pose2, PriorFactorPose2, Values
 
 
 def run(args: Namespace) -> None:
@@ -41,7 +36,7 @@ def run(args: Namespace) -> None:
     graph, initial = gtsam.readG2o(g2oFile)
 
     # Add prior on the pose having index (key) = 0
-    priorModel = gtsam.noiseModel.Diagonal.Variances(vector3(1e-6, 1e-6, 1e-8))
+    priorModel = gtsam.noiseModel.Diagonal.Variances(Point3(1e-6, 1e-6, 1e-8))
     graph.add(PriorFactorPose2(0, Pose2(), priorModel))
     print(graph)