diff --git a/wind_farm_gym/wind_map.py b/wind_farm_gym/wind_map.py
index ad73e6b..94a58d2 100644
--- a/wind_farm_gym/wind_map.py
+++ b/wind_farm_gym/wind_map.py
@@ -22,8 +22,8 @@ def __init__(self, size, dpi, cut_plane, turbines_raw_data, wind_speed_limits=No
         self.fig.subplots_adjust(top=1, bottom=0, right=1, left=0, hspace=0, wspace=0)
         self.ax = self.fig.gca()
         self.ax.set_axis_off()
-        self.ax.xaxis.set_major_locator(plt.NullLocator())
-        self.ax.yaxis.set_major_locator(plt.NullLocator())
+        self.ax.xaxis.set_visible(False)
+        self.ax.yaxis.set_visible(False)
         self.canvas = FigureCanvasAgg(self.fig)
         self.color_map = color_map
         if wind_speed_limits is None:
@@ -67,7 +67,7 @@ def render(self):
         self.draw()
 
     def clear(self):
-        self.ax.collections.clear()
+        self.ax.clear()
 
     def close(self):
         plt.close(self.fig)