diff --git a/Dockerfile b/Dockerfile
index edbaaee..1c13680 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,11 +12,9 @@ RUN mkdir /opt/local && \
     cd donkeycar && \
     mamba env create -f install/envs/ubuntu.yml && \
     eval "$(conda shell.bash hook)"
-RUN mamba install -n donkey cudatoolkit=10.1 tensorflow-gpu=2.2.0 -c anaconda -y && \
+RUN mamba install -n donkey cudatoolkit=10.1 tensorflow-gpu=2.2.0 -c anaconda nb_conda_kernels -y && \
     chown -R jovyan /home/jovyan
 
-
-
 RUN python -m ipykernel install --name=donkey --display-name="Donkey Car ($DONKEYCAR_VERSION-$DONKEYCAR_BRANCH)"
 RUN chown -R jovyan /opt/local