diff --git a/Project.toml b/Project.toml
index fcf0754..fb9ae22 100644
--- a/Project.toml
+++ b/Project.toml
@@ -18,7 +18,7 @@ SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 Makie = "ee78f7c6-11fb-53f2-987a-cfe4a2b5a57a"
 
 [extensions]
-MakieVizExt = "Makie"
+MakieExt = "Makie"
 
 [compat]
 BlockArrays = "0.16"
@@ -26,7 +26,7 @@ Colors = "0.12"
 GeometryBasics = "0.4.2"
 InfiniteArrays = "0.12, 0.13"
 LazySets = "1.54, 2"
-Makie = "0.17, 0.18, 0.19, 0.20"
+Makie = "0.17, 0.18, 0.19, 0.20, 0.21"
 Polyhedra = "0.7"
 julia = "1.7"
 
diff --git a/ext/MakieVizExt.jl b/ext/MakieExt.jl
similarity index 96%
rename from ext/MakieVizExt.jl
rename to ext/MakieExt.jl
index fdd59a3..b614e4b 100644
--- a/ext/MakieVizExt.jl
+++ b/ext/MakieExt.jl
@@ -1,4 +1,4 @@
-module MakieVizExt
+module MakieExt
 using TrajectoryGamesBase: TrajectoryGamesBase
 using Makie: Makie, @recipe
 using Colors: @colorant_str
@@ -6,7 +6,10 @@ using GeometryBasics: GeometryBasics
 
 Makie.plottype(::TrajectoryGamesBase.PolygonEnvironment) = Makie.Poly
 
-function Makie.convert_arguments(::Type{<:Makie.Poly}, environment)
+function Makie.convert_arguments(
+    ::Type{<:Makie.Poly},
+    environment::TrajectoryGamesBase.PolygonEnvironment,
+)
     geometry = GeometryBasics.Polygon(GeometryBasics.Point{2}.(environment.set.vertices))
     (geometry,)
 end
diff --git a/src/TrajectoryGamesBase.jl b/src/TrajectoryGamesBase.jl
index e86d456..b19bf90 100644
--- a/src/TrajectoryGamesBase.jl
+++ b/src/TrajectoryGamesBase.jl
@@ -65,7 +65,7 @@ export to_blockvector,
     unflatten_trajectory
 
 if !isdefined(Base, :get_extension)
-    include("../ext/MakieVizExt.jl")
+    include("../ext/MakieExt.jl")
 end
 
 end