Update xarray_backend.py

The "remove dimension" section should also have been updated

Author: jurjendejong

xarray_selafin/xarray_backend.py

@@ -249,9 +249,9 @@ def _raw_indexing_method(self, key):
             if plan_key is None:
                 data = data[:, 0]
             else:
-                data = data[:, 0, :]
+                data = data[:, :, 0]
         if isinstance(plan_key, int):
-            data = data[..., 0]
+            data = data[:, 0, :]
         if isinstance(time_key, int):
             data = data[0, ...]
         return data