diff --git a/utils/episode_artifacts.py b/utils/episode_artifacts.py
index 2a0e263..5503cba 100644
--- a/utils/episode_artifacts.py
+++ b/utils/episode_artifacts.py
@@ -110,7 +110,14 @@ def _plot_episode_heatmap(path: str, edge_rows: Sequence[Dict], edge_ids: Sequen
     ]
 
     for ax, (grid, cmap, title) in zip(axes, plots):
-        image = ax.imshow(np.ma.masked_invalid(grid), aspect="auto", origin="lower", cmap=cmap)
+        image = ax.imshow(
+            np.ma.masked_invalid(grid),
+            aspect="auto",
+            origin="lower",
+            cmap=cmap,
+            interpolation="nearest",
+            resample=False,
+        )
         ax.set_title(f"{title_prefix} {title}")
         ax.set_xlabel("Decision Step")
         ax.set_ylabel("Controlled Edge")