diff --git a/View/acoustic_inversion_tab.py b/View/acoustic_inversion_tab.py
index 9e84d3d..f702ede 100644
--- a/View/acoustic_inversion_tab.py
+++ b/View/acoustic_inversion_tab.py
@@ -1703,9 +1703,9 @@ class AcousticInversionTab(QWidget):
                     depth_data[k].shape[1]
                 )
 
-                r = np.zeros((
-                    depth_data[k].shape[1] * time_data[k].shape[1], 1
-                ))
+                r = np.zeros(
+                    depth_data[k].shape[1] * time_data[k].shape[1]
+                )
 
                 for i in range(time_data[k].shape[1]):
                     for j in range(depth_data[k].shape[1]):
@@ -1737,8 +1737,11 @@ class AcousticInversionTab(QWidget):
                     )
                 )
 
+            if os.path.splitext(filename)[1] != ".xlsx":
+                filename += ".xlsx"
+
             with pd.ExcelWriter(
-                    os.path.join(dirname, filename + '.xlsx')
+                    os.path.join(dirname, filename)
             ) as writer:
                 for k in range(self.combobox_acoustic_data_choice.count()):
                     results[k].to_excel(