diff --git a/utils/benchmark.py b/utils/benchmark.py
index e5b2268..5e0a2f8 100644
--- a/utils/benchmark.py
+++ b/utils/benchmark.py
@@ -56,7 +56,7 @@
         continue
 
     # Skip old BipedalWalker version
-    if 'Walker-v2' in trained_model or 'WalkerHardcore-v2' in trained_model:
+    if 'Walker-v2' in env_id or 'WalkerHardcore-v2' in env_id:
         continue
 
     reward_log = '{}/{}/'.format(args.benchmark_dir, trained_model)