diff --git a/src/device/Dockerfile b/src/device/Dockerfile
index 1f9e629bb12c90dd27b7d08fc7994e532cdb1267..4a86b79bac17973f8e3a7b5eceb549c0ef418a0a 100644
--- a/src/device/Dockerfile
+++ b/src/device/Dockerfile
@@ -86,11 +86,22 @@ COPY src/monitoring/__init__.py monitoring/__init__.py
 COPY src/monitoring/client/. monitoring/client/
 COPY src/device/. device/
 
+# Clone test mock tools
 RUN mkdir -p tests/tools/mock_ietf_actn_sdn_ctrl
 RUN touch tests/__init__.py
 RUN touch tests/tools/__init__.py
 RUN touch tests/tools/mock_ietf_actn_sdn_ctrl/__init__.py
 COPY src/tests/tools/mock_ietf_actn_sdn_ctrl/. tests/tools/mock_ietf_actn_sdn_ctrl/
 
+# Clone OpenConfig YANG models
+RUN mkdir -p /var/teraflow/device/service/drivers/gnmi_openconfig/git/openconfig/public
+RUN mkdir -p /tmp/openconfig
+RUN git clone https://github.com/openconfig/public.git /tmp/openconfig
+RUN git fetch
+RUN git checkout v4.4.0
+RUN mv /tmp/openconfig/release /var/teraflow/device/service/drivers/gnmi_openconfig/git/openconfig/public
+RUN mv /tmp/openconfig/third_party /var/teraflow/device/service/drivers/gnmi_openconfig/git/openconfig/public
+RUN rm -rf /tmp/openconfig
+
 # Start the service
 ENTRYPOINT ["python", "-m", "device.service"]