diff --git a/README.md b/README.md
index ae71a59c..39cf14d9 100644
--- a/README.md
+++ b/README.md
@@ -147,11 +147,11 @@ python tests/atari_init_test.py
 ```
 
 The entry point to the standard Atari 2600 experiment is
-[`dopamine/atari/train.py`](https://github.com/google/dopamine/blob/master/dopamine/atari/train.py).
+[`dopamine/discrete_domains/train.py`](https://github.com/google/dopamine/blob/master/dopamine/discrete_domains/train.py).
 To run the basic DQN agent,
 
 ```
-python -um dopamine.atari.train \
+python -um dopamine.discrete_domains.train \
   --agent_name=dqn \
   --base_dir=/tmp/dopamine \
   --gin_files='dopamine/agents/dqn/configs/dqn.gin'
diff --git a/dopamine/agents/rainbow/configs/c51_icml.gin b/dopamine/agents/rainbow/configs/c51_icml.gin
index 0cd5e309..e148591c 100644
--- a/dopamine/agents/rainbow/configs/c51_icml.gin
+++ b/dopamine/agents/rainbow/configs/c51_icml.gin
@@ -1,5 +1,4 @@
 # Hyperparameters used in Bellemare et al. (2017).
-import dopamine.atari.preprocessing
 import dopamine.agents.rainbow.rainbow_agent
 import dopamine.discrete_domains.atari_lib
 import dopamine.discrete_domains.run_experiment
diff --git a/dopamine/colab/agents.ipynb b/dopamine/colab/agents.ipynb
index 7d4f6302..b30267f8 100644
--- a/dopamine/colab/agents.ipynb
+++ b/dopamine/colab/agents.ipynb
@@ -76,7 +76,7 @@
         "import numpy as np\n",
         "import os\n",
         "from dopamine.agents.dqn import dqn_agent\n",
-        "from dopamine.atari import run_experiment\n",
+        "from dopamine.discrete_domains import run_experiment\n",
         "from dopamine.colab import utils as colab_utils\n",
         "from absl import flags\n",
         "\n",
@@ -411,4 +411,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/dopamine/colab/load_statistics.ipynb b/dopamine/colab/load_statistics.ipynb
index abd0b715..9d91858a 100644
--- a/dopamine/colab/load_statistics.ipynb
+++ b/dopamine/colab/load_statistics.ipynb
@@ -79,7 +79,7 @@
         "import numpy as np\n",
         "import os\n",
         "from dopamine.agents.dqn import dqn_agent\n",
-        "from dopamine.atari import run_experiment\n",
+        "from dopamine.discrete_domains import run_experiment\n",
         "from dopamine.colab import utils as colab_utils\n",
         "from absl import flags\n",
         "\n",
@@ -310,4 +310,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}