1 files changed, 11 insertions, 2 deletions
diff --git a/src/azure-cli/install.sh b/src/azure-cli/install.sh
index 6518f6c..de5d65d 100755
--- a/src/azure-cli/install.sh
+++ b/src/azure-cli/install.sh
@@ -9,6 +9,9 @@
 
 set -e
 
+# Clean up
+rm -rf /var/lib/apt/lists/*
+
 AZ_VERSION=${VERSION:-"latest"}
 
 MICROSOFT_GPG_KEYS_URI="https://packages.microsoft.com/keys/microsoft.asc"
@@ -44,7 +47,10 @@ apt_get_update()
 # Checks if packages are installed and installs them if not
 check_packages() {
     if ! dpkg -s "$@" > /dev/null 2>&1; then
-        apt_get_update
+        if [ "$(find /var/lib/apt/lists/* | wc -l)" = "0" ]; then
+            echo "Running apt-get update..."
+            apt-get update -y
+        fi
         apt-get -y install --no-install-recommends "$@"
     fi
 }
@@ -179,4 +185,7 @@ if [ "${use_pip}" = "true" ]; then
     fi
 fi
 
-echo "Done!"
-\ No newline at end of file
+# Clean up
+rm -rf /var/lib/apt/lists/*
+
+echo "Done!"