diff --git a/gtsam/sfm/sfm.i b/gtsam/sfm/sfm.i
index 26dc20c3e..c57e03174 100644
--- a/gtsam/sfm/sfm.i
+++ b/gtsam/sfm/sfm.i
@@ -329,7 +329,7 @@ class MatchIndicesMap {
   size_t size() const;
   bool empty() const;
   void clear();
-  gtsam::gtsfm::CorrespondenceIndices at(const pair<size_t, size_t>& keypair) const;
+  gtsam::gtsfm::CorrespondenceIndices at(const gtsam::IndexPair& keypair) const;
 };
 
 class Keypoints {
@@ -350,6 +350,7 @@ class KeypointsVector {
 gtsam::SfmTrack2dVector tracksFromPairwiseMatches(
     const gtsam::gtsfm::MatchIndicesMap& matches_dict,
     const gtsam::gtsfm::KeypointsVector& keypoints_list, bool verbose = false);
+
 }  // namespace gtsfm
 
 }  // namespace gtsam
diff --git a/python/gtsam/tests/test_DsfTrackGenerator.py b/python/gtsam/tests/test_DsfTrackGenerator.py
index e600227c9..284c3f5cc 100644
--- a/python/gtsam/tests/test_DsfTrackGenerator.py
+++ b/python/gtsam/tests/test_DsfTrackGenerator.py
@@ -7,7 +7,8 @@ import unittest
 
 import gtsam
 import numpy as np
-from gtsam import IndexPair, KeypointsVector, MatchIndicesMap, Point2, SfmMeasurementVector, SfmTrack2d
+from gtsam import (IndexPair, KeypointsVector, MatchIndicesMap, Point2,
+                   SfmMeasurementVector, SfmTrack2d)
 from gtsam.gtsfm import Keypoints
 from gtsam.utils.test_case import GtsamTestCase
 
@@ -84,12 +85,12 @@ class TestSfmTrack2d(GtsamTestCase):
     """Tests for SfmTrack2d."""
 
     def test_sfm_track_2d_constructor(self) -> None:
-        """ """
+        """Test construction of 2D SfM track."""
         measurements = SfmMeasurementVector()
         measurements.append((0, Point2(10, 20)))
         track = SfmTrack2d(measurements=measurements)
         track.measurement(0)
-        track.numberMeasurements() == 1
+        assert track.numberMeasurements() == 1
 
 
 if __name__ == "__main__":