diff --git a/rnalysis/utils/ontology.py b/rnalysis/utils/ontology.py
index 9a52b260..ea9bfdb6 100644
--- a/rnalysis/utils/ontology.py
+++ b/rnalysis/utils/ontology.py
@@ -717,7 +717,7 @@ def plot_ontology(self, namespace: Literal[param_typing.GO_ASPECTS], results_df:
         sm = ScalarMappable(cmap=colormap, norm=plt.Normalize(*bounds))
         sm.set_array(np.array([]))
         cbar_label_kwargs = dict(label=ylabel, fontsize=16, labelpad=15)
-        cbar = fig.colorbar(sm, ticks=range(int(bounds[0]), int(bounds[1]) + 1), location='bottom')
+        cbar = fig.colorbar(sm, ticks=range(int(bounds[0]), int(bounds[1]) + 1), location='bottom', ax=ax)
         cbar.set_label(**cbar_label_kwargs)
         cbar.ax.tick_params(labelsize=14, pad=6)