[gui][shift] Keep filter after correcting shift

darfix/gui/shiftCorrectionWidget.py

@@ -223,10 +223,7 @@ class ShiftCorrectionWidget(qt.QMainWindow):
         if dataset is None:
             dataset = self.dataset
         nframe = self._sv.getFrameNumber()
-        if self.indices is None:
-            self._sv.setStack(dataset.get_data() if dataset is not None else None)
-        else:
-            self._sv.setStack(dataset.get_data(self.indices) if dataset is not None else None)
+        self._sv.setStack(dataset.get_data(self.indices, self._dimension) if dataset is not None else None)
         self._sv.setFrameNumber(nframe)
 
     def clearStack(self):