Update functions/PitchPlotFunctions.py

functions/PitchPlotFunctions.py

@@ -616,12 +616,12 @@ class PitchPlotFunctions:
         ax_footer.axis('off')
         ax_legend.axis('off')
                 # Add clip path to constrain brush
-        for ax in fig.get_axes():
-            if ax.get_xlabel() == 'Horizontal Break (in)':  # Identify the main plot
-                ax.set_clip_on(True)
-                ax.patch.set_clip_on(True)
-                for artist in ax.get_children():
-                    artist.set_clip_on(True)
+
+        ax_main_plot.set_clip_on(True)
+        ax_main_plot.patch.set_clip_on(True)
+        for artist in ax_main_plot.get_children():
+            artist.set_clip_on(True)
+
         # Adjust layout and show the figure
         # fig.tight_layout()
         fig.subplots_adjust(left=0.01, right=0.99, top=0.99, bottom=0.01)