diff --git a/Dockerfile b/Dockerfile
index b11b41a..b322fad 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -19,19 +19,19 @@ RUN AZ_REPO=$(lsb_release -cs) && \
     tee /etc/apt/sources.list.d/azure-cli.list && \
     curl -L https://packages.microsoft.com/keys/microsoft.asc | apt-key add -
 
-RUN curl https://packages.microsoft.com/config/ubuntu/18.04/prod.list > ./microsoft-prod.list && \
+RUN curl https://packages.microsoft.com/config/ubuntu/18.04/multiarch/prod.list > ./microsoft-prod.list && \
     cp ./microsoft-prod.list /etc/apt/sources.list.d/ && \
     curl https://packages.microsoft.com/keys/microsoft.asc | gpg --dearmor > microsoft.gpg && \
-    cp ./microsoft.gpg /etc/apt/trusted.gpg.d/ 
+    cp ./microsoft.gpg /etc/apt/trusted.gpg.d/
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
     azure-cli \
     moby-cli \
-    moby-engine && \ 
-    apt-get install -y --no-install-recommends iotedge=1.0.0-1 && \ 
+    moby-engine && \
+    apt-get install -y --no-install-recommends iotedge=1.1.1-1 && \
     rm -rf /var/lib/apt/lists/*
-    
-RUN az extension add --name azure-cli-iot-ext
+
+RUN az extension add --name azure-iot
 
 COPY edge-provision.sh /usr/local/bin/
 RUN chmod +x /usr/local/bin/edge-provision.sh
diff --git a/edge-provision.sh b/edge-provision.sh
index e1e7139..0a987ec 100644
--- a/edge-provision.sh
+++ b/edge-provision.sh
@@ -26,7 +26,7 @@ agent:
   type: "docker"
   env: {}
   config:
-    image: "mcr.microsoft.com/azureiotedge-agent:1.0"
+    image: "mcr.microsoft.com/azureiotedge-agent:1.1"
     auth: {}
 hostname: $(cat /proc/sys/kernel/hostname)
 connect:
@@ -43,7 +43,7 @@ EOF
 
 cat /etc/iotedge/config.yaml
 
-iotedged -c /etc/iotedge/config.yaml 
+iotedged -c /etc/iotedge/config.yaml
 
 }
 
@@ -64,5 +64,5 @@ done
 if [ -z "$connectionString" ]; then
     echo "No connectionString provided, provisioning as a brand new IoTEdge device with name: $(hostname)"
     provision
-fi 
+fi
 startEdgeRuntime