diff --git a/deployment/dockerfiles/Dockerfile b/deployment/dockerfiles/Dockerfile
index f0d92b916..a67f89794 100644
--- a/deployment/dockerfiles/Dockerfile
+++ b/deployment/dockerfiles/Dockerfile
@@ -73,6 +73,9 @@ ENTRYPOINT ["/bin/bash"]
 FROM prepare-compile-enclave AS compile-tendermint-enclave
 
 ARG SGX_MODE=SW
+ARG FEATURES_U
+
+ARG SGX_MODE=${SGX_MODE}
 ENV FEATURES_U=${FEATURES_U}
 
 RUN git clone --branch main https://github.com/scrtlabs/tm-secret-enclave.git