diff --git a/enjoy_mineral_shards.py b/enjoy_mineral_shards.py
index 1d1ad49..217af5b 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