From 7e596d4dad3793d6430204b7a9b214a9dc39616c Mon Sep 17 00:00:00 2001
From: cako <c.dacostaf@gmail.com>
Date: Sun, 23 Jan 2022 17:44:35 -0500
Subject: [PATCH] minor: Remove all remaining `fftshift`

---
 pylops/waveeqprocessing/oneway.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/pylops/waveeqprocessing/oneway.py b/pylops/waveeqprocessing/oneway.py
index d5b7677e..4f446348 100644
--- a/pylops/waveeqprocessing/oneway.py
+++ b/pylops/waveeqprocessing/oneway.py
@@ -166,10 +166,17 @@ def PhaseShift(vel, dz, nt, freq, kx, ky=None, dtype="float64"):
         dims = (nt, kx.size, ky.size)
         dimsfft = (freq.size, kx.size, ky.size)
     Fop = FFT(dims, dir=0, nfft=nt, real=True, dtype=dtype)
-    Kxop = FFT(dimsfft, dir=1, nfft=kx.size, real=False, fftshift=True, dtype=dtypefft)
+    Kxop = FFT(
+        dimsfft, dir=1, nfft=kx.size, real=False, fftshift_after=True, dtype=dtypefft
+    )
     if ky is not None:
         Kyop = FFT(
-            dimsfft, dir=2, nfft=ky.size, real=False, fftshift=True, dtype=dtypefft
+            dimsfft,
+            dir=2,
+            nfft=ky.size,
+            real=False,
+            fftshift_after=True,
+            dtype=dtypefft,
         )
     Pop = _PhaseShift(vel, dz, freq, kx, ky, dtypefft)
     if ky is None: