diff --git a/Dockerfile.tchem b/Dockerfile.tchem
index b34de0eb8..0bb7453d5 100644
--- a/Dockerfile.tchem
+++ b/Dockerfile.tchem
@@ -28,6 +28,7 @@ RUN apt-get update \
 
 RUN git clone https://github.com/PCLAeroParams/TChem-atm.git /tchem_dir/
 WORKDIR /tchem_dir/
+RUN git checkout -b add_aerosol_driver origin/add_aerosol_driver
 RUN git submodule update --init --recursive
 
 RUN cmake -S /tchem_dir/external/Tines/ext/kokkos -B /build/kokkos_build \