diff --git a/Dockerfile/agent/Dockerfile b/Dockerfile/agent/Dockerfile
index 5edb1895..856d6a07 100644
--- a/Dockerfile/agent/Dockerfile
+++ b/Dockerfile/agent/Dockerfile
@@ -1,14 +1,10 @@
 FROM ubuntu:focal
 LABEL version=0.1.4
 
-RUN apt-get update -y && apt-get install -y ssh
-#RUN mkdir -p root/.ssh/ && \
-#    cd root/.ssh && \
-#    echo $'Host * \n\
-#        StrictHostKeyChecking no \n\
-#        UserKnownHostsFile /dev/null \n '\
-#    > config
-
+RUN apt-get update -y && \
+    apt-get install -y ssh && \
+    apt-get -y clean && \
+    rm -rf /var/lib/apt/lists/*
 
 COPY ./klevr  /
 COPY ./entrypoint.sh  /