diff --git a/nethack_baselines/torchbeast/Dockerfile b/nethack_baselines/torchbeast/Dockerfile
index 13e32b50de46038c1ac4a0f3759af332d136f12a..eb2897aff4765a4f753ef2ab0d70f1897eca7f45 100644
--- a/nethack_baselines/torchbeast/Dockerfile
+++ b/nethack_baselines/torchbeast/Dockerfile
@@ -1,7 +1,7 @@
 # -*- mode: dockerfile -*-
 FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04
 
-ARG PYTHON_VERSION=3.8
+ARG PYTHON_VERSION=3.7
 ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update && apt-get install -yq \
@@ -43,7 +43,7 @@ RUN python -c 'import torch'
 
 # Install TorchBeast
 WORKDIR /opt/
-RUN git clone https://github.com/condnsdmatters/torchbeast.git --branch eric/experimental-port  --recursive
+RUN git clone https://github.com/facebookresearch/torchbeast.git --recursive
 
 WORKDIR /opt/torchbeast
 RUN pip install -r requirements.txt