diff --git a/src/napari_sam_labeling_tools/_sam_prompt_segmentation_widget.py b/src/napari_sam_labeling_tools/_sam_prompt_segmentation_widget.py
index 9403b9b..6c2e579 100644
--- a/src/napari_sam_labeling_tools/_sam_prompt_segmentation_widget.py
+++ b/src/napari_sam_labeling_tools/_sam_prompt_segmentation_widget.py
@@ -642,5 +642,5 @@ def export_segmentation(self, out_format="nrrd"):
             self, "Jug Lab", ".", "Segmentation(*.nrrd)"
         )
         if selected_file is not None and len(selected_file) > 0:
-            nrrd.write(selected_file, self.segmentation_layer.data)
-            notif.show_info("Selected segmentation was saved successfully.")
+            nrrd.write(selected_file, np.transpose(self.segmentation_layer.data))
+            notif.show_info("Selected segmentation was exported successfully.")
diff --git a/src/napari_sam_labeling_tools/_sam_rf_segmentation_widget.py b/src/napari_sam_labeling_tools/_sam_rf_segmentation_widget.py
index 03b3c5a..1d21827 100644
--- a/src/napari_sam_labeling_tools/_sam_rf_segmentation_widget.py
+++ b/src/napari_sam_labeling_tools/_sam_rf_segmentation_widget.py
@@ -659,5 +659,5 @@ def export_segmentation(self, out_format="nrrd"):
         if selected_file is not None and len(selected_file) > 0:
             if not selected_file.endswith(".nrrd"):
                 selected_file += ".nrrd"
-            nrrd.write(selected_file, self.segmentation_layer.data)
-            notif.show_info("Selected segmentation was saved successfully.")
+            nrrd.write(selected_file, np.transpose(self.segmentation_layer.data))
+            notif.show_info("Selected segmentation was exported successfully.")