diff --git a/enjoy_mineral_shards.py b/enjoy_mineral_shards.py
index 1d1ad49..48b8d9c 100644
--- a/enjoy_mineral_shards.py
+++ b/enjoy_mineral_shards.py
@@ -1,7 +1,7 @@
 import sys
 
 import baselines.common.tf_util as U
-import gflags as flags
+from absl import flags
 import numpy as np
 from baselines import deepq
 from pysc2.env import environment
@@ -30,8 +30,9 @@
 
 def main():
   FLAGS(sys.argv)
+  #with sc2_env.SC2Env()
   with sc2_env.SC2Env(
-      "CollectMineralShards",
+      map_name="CollectMineralShards",
       step_mul=step_mul,
       visualize=True,
       game_steps_per_episode=steps * step_mul) as env:
diff --git a/train_mineral_shards.py b/train_mineral_shards.py
index 79465a5..f93f9c1 100644
--- a/train_mineral_shards.py
+++ b/train_mineral_shards.py
@@ -1,6 +1,6 @@
 import sys
 
-import gflags as flags
+from absl import flags
 from baselines import deepq
 from pysc2.env import sc2_env
 from pysc2.lib import actions
@@ -19,7 +19,7 @@
 def main():
   FLAGS(sys.argv)
   with sc2_env.SC2Env(
-      "CollectMineralShards",
+      map_name="CollectMineralShards",
       step_mul=step_mul,
       visualize=True) as env: