diff --git a/docs/contributor/releases.md b/docs/contributor/releases.md
index 4d385c83..eaf795c7 100644
--- a/docs/contributor/releases.md
+++ b/docs/contributor/releases.md
@@ -39,7 +39,7 @@ Otherwise **`cz bump`** can keep treating breaking changes on **0.x** as minor b
 
 ## 3. Update the contributor guide
 
-Delete this file and update {doc}`./index` to remove the `going-stable-1-0`toctree entry.
+Delete this file and update {doc}`./index` to remove the `going-stable-1-0` toctree entry.
 
 ## 4. Cutting 1.0.0
 
diff --git a/docs/index.md b/docs/index.md
index 728886aa..3c403e99 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -13,7 +13,7 @@ as defined in [Towards the certification of AI-based systems](https://doi.org/10
 
 ## Where does it fit in my workflow?
 
-RAITAP is configured via YAML [Hydra](https://hydra.cc/) configs or CLI flags, and then ran via a CLI command.
+RAITAP is configured via YAML [Hydra](https://hydra.cc/) configs or CLI flags, and then run via a CLI command.
 
 This means it can be used either as:
 
diff --git a/docs/using-raitap/installation.md b/docs/using-raitap/installation.md
index e0430f94..643d87ed 100644
--- a/docs/using-raitap/installation.md
+++ b/docs/using-raitap/installation.md
@@ -15,7 +15,7 @@ pip install raitap
 ```
 
 :::{note}
-RAITAP was currently tested with Python 3.13.x. Ensure your project matches this requirement, or expect possible issues.
+RAITAP is currently tested with Python 3.13.x. Ensure your project matches this requirement, or expect possible issues.
 :::
 
 ## 2. Install optional dependencies
diff --git a/src/raitap/transparency/visualisers/tests/test_visualisers.py b/src/raitap/transparency/visualisers/tests/test_visualisers.py
index ca39dd22..7c6ea5e4 100644
--- a/src/raitap/transparency/visualisers/tests/test_visualisers.py
+++ b/src/raitap/transparency/visualisers/tests/test_visualisers.py
@@ -40,7 +40,7 @@ def test_visualise_tensor(self, sample_images: torch.Tensor) -> None:
 
         fig = visualiser.visualise(attributions)
         assert fig is not None
-        assert len(fig.axes) >= 4  # at least one axes per sample
+        assert len(fig.axes) == 8  # 2 axes per sample (original + attribution) for 4 samples
 
     @pytest.mark.usefixtures("needs_captum")
     def test_max_samples_limit(self) -> None:
@@ -48,7 +48,7 @@ def test_max_samples_limit(self) -> None:
         large_batch = torch.randn(64, 3, 32, 32)
 
         fig = visualiser.visualise(large_batch, max_samples=4)
-        assert len(fig.axes) >= 4
+        assert len(fig.axes) == 8
 
     @pytest.mark.usefixtures("needs_captum")
     def test_overlay_with_inputs(self, sample_images: torch.Tensor) -> None:
@@ -81,7 +81,7 @@ def test_can_disable_original_image_panel(self, sample_images: torch.Tensor) ->
 
         fig = visualiser.visualise(attributions, inputs=sample_images, max_samples=2)
 
-        assert len(fig.axes) >= 2
+        assert len(fig.axes) == 2
         assert fig.axes[0].get_title() == ""
 
     @pytest.mark.usefixtures("needs_captum")