diff --git a/pyrost/data_processing.py b/pyrost/data_processing.py
index 60eae85b60a73dcf1940078159149ca84cb6a1e6..ae6d90b60b672b0bf0c3c64a8c10ae1f2aff80a5 100755
--- a/pyrost/data_processing.py
+++ b/pyrost/data_processing.py
@@ -152,8 +152,8 @@ class STData(DataContainer):
         # Initialize a list of AbberationsFit objects
         self._ab_fits = []
         if self._isphase:
-            self._ab_fits.extend([AbberationsFit.import_data(self, 0),
-                                  AbberationsFit.import_data(self, 1)])
+            self._ab_fits.extend([AbberationsFit.import_data(self, axis=0),
+                                  AbberationsFit.import_data(self, axis=1)])
 
     @property
     def _isdefocus(self):
@@ -350,8 +350,8 @@ class STData(DataContainer):
 
             # Initialize AbberationsFit objects
             self._ab_fits.clear()
-            self._ab_fits.extend([AbberationsFit.import_data(self, 0),
-                                  AbberationsFit.import_data(self, 1)])
+            self._ab_fits.extend([AbberationsFit.import_data(self, axis=0),
+                                  AbberationsFit.import_data(self, axis=1)])
             return self
         else:
             raise ValueError("the SpeckleTracking object doesn't belong to the data container")