diff --git a/docs/_scripts/generate_gifs.py b/docs/_scripts/generate_gifs.py
index 7183739e0..6373ab741 100644
--- a/docs/_scripts/generate_gifs.py
+++ b/docs/_scripts/generate_gifs.py
@@ -1,6 +1,5 @@
 import subprocess
-
-from pettingzoo.test.all_modules import all_environments
+from test.all_modules import all_environments
 
 # procs = []
 for name, module in all_environments.items():
diff --git a/docs/_scripts/graph_gen.py b/docs/_scripts/graph_gen.py
index 44d197246..14bfd72e4 100644
--- a/docs/_scripts/graph_gen.py
+++ b/docs/_scripts/graph_gen.py
@@ -2,8 +2,7 @@
 import os
 import subprocess
 from collections import defaultdict
-
-from pettingzoo.test.all_modules import all_environments
+from test.all_modules import all_environments
 
 
 def generate_cycle_words(agents, is_classic):
diff --git a/docs/_scripts/text_capture.py b/docs/_scripts/text_capture.py
index 825d3992b..c1ccc1db6 100644
--- a/docs/_scripts/text_capture.py
+++ b/docs/_scripts/text_capture.py
@@ -1,10 +1,9 @@
 import json
 import random
+from test.all_modules import all_environments
 
 import numpy as np
 
-from pettingzoo.test.all_modules import all_environments
-
 
 def shrink_lines(data, max_height, max_width):
     lines = data.split("\r\n")[:max_height]
@@ -17,7 +16,6 @@ def shrink_lines(data, max_height, max_width):
     if "classic/" not in name:
         continue
     env = module.env()
-    # env = gin_rummy_v0.env()
     env.reset()
 
     all_text = [env.render()]