diff --git a/pufferlib/config/cogames.ini b/pufferlib/config/cogames.ini
index 674b48e2e6..eb0679ae77 100644
--- a/pufferlib/config/cogames.ini
+++ b/pufferlib/config/cogames.ini
@@ -1,6 +1,6 @@
 [base]
 package = cogames
-env_name = cogames.cogs_v_clips.training_facility.harvest cogames.cogs_v_clips.training_facility.assemble cogames.cogs_v_clips.machina_1.open_world
+env_name = basic aligner_tutorial miner_tutorial scout_tutorial scrambler_tutorial energy_starved
 policy_name = Policy
 rnn_name = Recurrent
 
@@ -12,7 +12,7 @@ zero_copy = True
 
 [env]
 render_mode = none
-variants = heart_chorus inventory_heart_tune
+variants = standard
 
 [train]
 total_timesteps = 50_000_000
diff --git a/pufferlib/environments/cogames/environment.py b/pufferlib/environments/cogames/environment.py
index 0fbe475957..6ac90c72d5 100644
--- a/pufferlib/environments/cogames/environment.py
+++ b/pufferlib/environments/cogames/environment.py
@@ -21,7 +21,6 @@ def make(name="cogames.cogs_v_clips.machina_1.open_world", variants=None, cogs=N
     simulator = Simulator()
     simulator.add_event_handler(StatsTracker(NoopStatsWriter()))
     env = PufferMettaGridEnv(simulator=simulator, cfg=env_cfg, buf=buf, seed=seed or 0)
-    env.render_mode = render
     if seed:
         env.reset(seed)
     return env
diff --git a/pyproject.toml b/pyproject.toml
index 35d91eaacb..6b54a7d304 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -128,7 +128,6 @@ cogames = [
     'hydra-core',
     'duckdb',
     'raylib>=5.5.0',
-    'mettagrid @ git+https://github.com/metta-ai/mettagrid.git',
     'cogames @ git+https://github.com/metta-ai/cogames.git',
 ]
 
diff --git a/setup.py b/setup.py
index 8a8444dd61..2b2485878e 100644
--- a/setup.py
+++ b/setup.py
@@ -272,7 +272,7 @@ def run(self):
 
 install_requires = [
     'setuptools',
-    'numpy<2.0',
+    'numpy',
     'shimmy[gym-v21]',
     'gym==0.23',
     'gymnasium>=0.29.1',