diff --git a/dockers/docker-ptf/Dockerfile.j2 b/dockers/docker-ptf/Dockerfile.j2
index 3304ff643bb8..899bba45c47f 100644
--- a/dockers/docker-ptf/Dockerfile.j2
+++ b/dockers/docker-ptf/Dockerfile.j2
@@ -84,6 +84,7 @@ RUN rm -rf /debs \
     && pip install paramiko \
     && pip install flask   \
     && pip install exabgp==3.4.17\
+    && pip install pyaml   \
     && mkdir -p /opt       \
     && cd /opt             \
     && wget https://raw.githubusercontent.com/p4lang/ptf/master/ptf_nn/ptf_nn_agent.py