diff --git a/py4DSTEM/tomography/tomography.py b/py4DSTEM/tomography/tomography.py
index ac0bfdb23..7240fe1af 100644
--- a/py4DSTEM/tomography/tomography.py
+++ b/py4DSTEM/tomography/tomography.py
@@ -568,7 +568,7 @@ def _solve_for_diffraction_pattern_centering(
             robust_thresh=robust_thresh,
         )
 
-        return qx0_fit.data, qy0_fit.data
+        return qx0_fit, qy0_fit
 
     def _reshape_diffraction_patterns(
         self,