diff --git a/templates/lib/prism/dot_visitor.rb.erb b/templates/lib/prism/dot_visitor.rb.erb
index e9c81e4545..cd2998fe61 100644
--- a/templates/lib/prism/dot_visitor.rb.erb
+++ b/templates/lib/prism/dot_visitor.rb.erb
@@ -1,4 +1,5 @@
-require "cgi"
+require "cgi/escape"
+require "cgi/util" unless defined?(CGI::EscapeExt)
 
 module Prism
   # This visitor provides the ability to call Node#to_dot, which converts a