diff --git a/sims/Timeloop/Dockerfile b/sims/Timeloop/Dockerfile
index aa2292ad..9b1eb3d6 100644
--- a/sims/Timeloop/Dockerfile
+++ b/sims/Timeloop/Dockerfile
@@ -30,13 +30,13 @@ CMD conda env create -f environment.yml
 RUN apt-get update && apt-get -y install libgmp-dev gcc g++
   
 # Make RUN commands use the new environment:
-RUN echo "conda activate oss-arch-gym" >> ~/.bashrc
-CMD conda run -n oss-arch-gym
+RUN echo "conda activate arch-gym" >> ~/.bashrc
+CMD conda run -n arch-gym
 SHELL ["/bin/bash", "--login", "-c"]
 
 RUN cd oss-arch-gym/acme && pip install .[jax,tf,testing,envs]
-ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/bin/conda/envs/oss-arch-gym/lib
+ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/bin/conda/envs/arch-gym/lib
 RUN echo $LD_LIBRARY_PATH
 
-ENTRYPOINT ["conda", "run", "-n", "oss-arch-gym", "python", "/home/workspace/src/timeloop-examples/workspace/final-project/example_designs/eyeriss_like/arch-gym/sims/Timeloop/launch_gcp.py"]
+ENTRYPOINT ["conda", "run", "-n", "arch-gym", "python", "/home/workspace/src/timeloop-examples/workspace/final-project/example_designs/eyeriss_like/arch-gym/sims/Timeloop/launch_gcp.py"]