diff --git a/python/gtsam/tests/test_Pose2.py b/python/gtsam/tests/test_Pose2.py
index f18eab079..213de655e 100644
--- a/python/gtsam/tests/test_Pose2.py
+++ b/python/gtsam/tests/test_Pose2.py
@@ -13,7 +13,7 @@ import unittest
 import numpy as np
 
 import gtsam
-from gtsam import Point2, Point2Pairs, Pose2
+from gtsam import align, Point2, Point2Pairs, Pose2
 from gtsam.utils.test_case import GtsamTestCase
 
 
@@ -58,7 +58,7 @@ class TestPose2(GtsamTestCase):
 
         # fmt: on
         ab_pairs = Point2Pairs(list(zip(pts_a, pts_b)))
-        bTa = Pose2.align(ab_pairs)
+        bTa = align(ab_pairs)
         aTb = bTa.inverse()
         assert aTb is not None