diff --git a/docs/conf.py b/docs/conf.py
index 2ef7aed..88f36c9 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -43,7 +43,7 @@
 # The short X.Y version
 version = '0.9'
 # The full version, including alpha/beta/rc tags
-release = '0.9.0'
+release = '0.9.1'
 
 # -- General configuration ---------------------------------------------------
 
diff --git a/neptunecontrib/versioning/data.py b/neptunecontrib/versioning/data.py
index c8b9810..05a6e3f 100644
--- a/neptunecontrib/versioning/data.py
+++ b/neptunecontrib/versioning/data.py
@@ -227,7 +227,8 @@ def _get_filepaths(dir_path):
 
 def _get_collated_image(filepaths, label=None, figsize=(16, 12), title_size=30):
     n = len(filepaths)
-    yn, xn = int(np.floor(np.sqrt(n))), int(np.ceil(np.sqrt(n)))
+    yn = int(np.floor(np.sqrt(n)))
+    xn = int(np.ceil(n / yn))
 
     fig, axs = plt.subplots(yn, xn, figsize=figsize)
     fig.suptitle(label, fontsize=title_size)
diff --git a/setup.py b/setup.py
index faeb7fb..e86d175 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@ def main():
         requirements = [r.strip() for r in f]
         setup(
             name='neptune-contrib',
-            version='0.9.0',
+            version='0.9.1',
             description='Neptune Python library contributions',
             author='neptune.ml',
             author_email='contact@neptune.ml',