diff --git a/src/container.rs b/src/container.rs
index 5c4ee6f..7c2fa83 100644
--- a/src/container.rs
+++ b/src/container.rs
@@ -28,7 +28,7 @@ pub fn deploy_enclave(
     // --storage-opt size={disk_mb}m
 
     let docker_deploy_str = format!(
-        r#"docker run -d --name {container_name_uuid} --memory={memory_mb}m --cpus={vcpu} \
+        r#"docker run -d --restart unless-stopped --name {container_name_uuid} --memory={memory_mb}m --cpus={vcpu} \
         -v {enclave_path}:/enclave_package --device /dev/sgx/enclave --device /dev/sgx/provision \
         {port_maping_string} noormohammedb/occlum-enclave:v1 {hratls_pubkey}"#
     );