diff --git a/py4DSTEM/process/diffraction/flowlines.py b/py4DSTEM/process/diffraction/flowlines.py
index 0dc3cb7d7..5af64f73e 100644
--- a/py4DSTEM/process/diffraction/flowlines.py
+++ b/py4DSTEM/process/diffraction/flowlines.py
@@ -16,7 +16,7 @@
 
 
 def make_orientation_histogram(
-    bragg_peaks: PointListArray = None,
+    bragg_peaks=None,
     radial_ranges: np.ndarray = None,
     orientation_map=None,
     orientation_ind: int = 0,
@@ -38,7 +38,7 @@ def make_orientation_histogram(
     texture OrientationMap generated by the ACOM module of py4DSTEM.
 
     Args:
-        bragg_peaks (PointListArray):       2D of pointlists containing centered peak locations.
+        bragg_peaks (BraggVectors):         bragg_vectos containing centered peak locations.
         radial_ranges (np array):           Size (N x 2) array for N radial bins, or (2,) for a single bin.
         orientation_map (OrientationMap):   Class containing the Euler angles to generate a flowline map.
         orientation_ind (int):              Index of the orientation map (default 0)
@@ -110,7 +110,7 @@ def make_orientation_histogram(
             add_data = False
 
             if orientation_map is None:
-                p = bragg_peaks.get_pointlist(rx, ry)
+                p = bragg_peaks.cal[rx, ry]
                 r2 = p.data["qx"] ** 2 + p.data["qy"] ** 2
                 sub = np.logical_and(
                     r2 >= radial_ranges_2[a0, 0], r2 < radial_ranges_2[a0, 1]