diff --git a/assets/state-mig-manager/0420_configmap.yaml b/assets/state-mig-manager/0420_configmap.yaml
index aec665dc4f..2ae17ad13c 100644
--- a/assets/state-mig-manager/0420_configmap.yaml
+++ b/assets/state-mig-manager/0420_configmap.yaml
@@ -14,13 +14,13 @@ data:
       echo "waiting for the driver validations to be ready..."
       sleep 5
     done
-    
+
     set -o allexport
     cat /run/nvidia/validations/driver-ready
     . /run/nvidia/validations/driver-ready
       
     # manually export additional envs required by mig-manager
-    export WITH_SHUTDOWN_HOST_GPU_CLIENTS=$IS_HOST_DRIVER
+    export WITH_SHUTDOWN_HOST_GPU_CLIENTS=${WITH_SHUTDOWN_HOST_GPU_CLIENTS:-$IS_HOST_DRIVER}
     echo "WITH_SHUTDOWN_HOST_GPU_CLIENTS=$WITH_SHUTDOWN_HOST_GPU_CLIENTS"
 
     echo "Starting nvidia-mig-manager"