diff --git a/datahandling/datahandling_interface.py b/datahandling/datahandling_interface.py
index e2fde4f28b767dead04a22a9a7d57e71124c53a2..f8c21ed01ea7881a98f7fba0f8316df93056d8c9 100644
--- a/datahandling/datahandling_interface.py
+++ b/datahandling/datahandling_interface.py
@@ -152,7 +152,7 @@ class DataHandling(ABC):
         """
 
     @abstractmethod
-    def swap(self, name1, name2):
+    def swap(self, name1, name2, gpu=False):
         """Swaps data of two arrays"""
 
     # ------------------------------- CPU/GPU transfer -----------------------------------------------------------------