diff --git a/Dockerfile b/Dockerfile
index 3c83525ec925e..b9a56e67e8d7b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -172,7 +172,7 @@ RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \
     && python3 --version
 
 RUN apt-get update -y \
-    && apt-get install -y python3-pip git curl libibverbs-dev
+    && apt-get install -y python3-pip git vim curl libibverbs-dev
 
 # Install pip s.t. it will be compatible with our PYTHON_VERSION
 RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python${PYTHON_VERSION}