diff --git a/pes_to_spec/__init__.py b/pes_to_spec/__init__.py
index 439057e6c3c9215cc40f5848237329de2dbbc43d..26f327e8cd518bddd24ff553e14bbf7c935b8735 100644
--- a/pes_to_spec/__init__.py
+++ b/pes_to_spec/__init__.py
@@ -2,4 +2,4 @@
 Estimate high-resolution photon spectrometer data from low-resolution non-invasive measurements.
 """
 
-VERSION = "0.1.0"
+VERSION = "0.1.1"
diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py
index 725e1e786e2c0ad6bfec7e6e7256bc063b027bb4..78d944c2b323e7481c11a1d78bbdddd88c039f32 100644
--- a/pes_to_spec/model.py
+++ b/pes_to_spec/model.py
@@ -846,7 +846,7 @@ class Model(TransformerMixin, BaseEstimator):
         G = np.conjugate(H) * S / (np.absolute(H)**2 * S + N)
         # generate a gaussian
         gaussian = np.exp(-0.5*(e_axis)**2/self.high_res_sigma**2)
-        gaussian /= np.sum(gaussian, axis=1, keepdims=True)
+        gaussian /= np.sum(gaussian, axis=0, keepdims=True)
         G *= gaussian
         self.wiener_filter = np.fft.ifft(G)
         self.wiener_filter_ft = G