diff --git a/CHANGELOG.md b/CHANGELOG.md
new file mode 100644
index 0000000000000000000000000000000000000000..af278493ef973c9fa6baf2e4f6a09ab1ed3e6456
--- /dev/null
+++ b/CHANGELOG.md
@@ -0,0 +1,10 @@
+# Changelog
+
+## v0.1.1
+
+- solve issue when exporting in PNG
+
+
+## v0.1.0 - 2023-10-17
+
+Initial version
\ No newline at end of file
diff --git a/pesto/export.py b/pesto/export.py
index daab6a96f5d474f18e75cb914c3e7171ccdc3a49..05c7c5af93008a5e6c57a301ff621c14b740a0c9 100644
--- a/pesto/export.py
+++ b/pesto/export.py
@@ -41,7 +41,7 @@ def export_png(output_file: str, timesteps, confidence, activations, lims=(21, 1
 
     bps = activations.shape[1] // 128
     activations = activations[:, bps*lims[0]: bps*lims[1]]
-    activations = activations * confidence.unsqueeze(1)
+    activations = activations * confidence[:, None]
     plt.imshow(activations.T,
                aspect='auto', origin='lower', cmap='inferno',
                extent=(timesteps[0], timesteps[-1]) + lims)