diff --git a/provision-device b/provision-device
index a1fb8b8..30c2dd2 100755
--- a/provision-device
+++ b/provision-device
@@ -1,30 +1,30 @@
-#!/bin/sh
+#!/bin/bash
 
-PARTITION_BIN_SRC=${ANDROID_BUILD_TOP}/vendor/bsp/qcom/device/dragonboard/qcom_410c_non-hlos_binaries_A8016BAAAAGLYD2131.1/partitions
-VENDOR_PARTITION_BIN_SRC=${ANDROID_BUILD_TOP}/vendor/bsp/qcom/device/dragonboard/qcom_410c_non-hlos_binaries_A8016BAAAAGLYD2131.1/vendor_partitions
+OS=${ANDROID_PROVISION_OS_PARTITIONS:-${ANDROID_PRODUCT_OUT}}
+VENDOR=${ANDROID_PROVISION_VENDOR_PARTITIONS:-${ANDROID_BUILD_TOP}/vendor/bsp/qcom/device/dragonboard/qcom_410c_non-hlos_binaries_A8016BAAAAGLYD2131.1/}
 
 fastboot \
-  flash partition ${PARTITION_BIN_SRC}/partition-tables.bin \
-  flash modem ${VENDOR_PARTITION_BIN_SRC}/modem.bin \
-  flash fsc ${VENDOR_PARTITION_BIN_SRC}/fsc.bin \
-  flash sbl1 ${VENDOR_PARTITION_BIN_SRC}/sbl1.bin \
-  flash sbl1bak ${VENDOR_PARTITION_BIN_SRC}/sbl1.bin \
-  flash rpm ${VENDOR_PARTITION_BIN_SRC}/rpm.bin \
-  flash rpmbak ${VENDOR_PARTITION_BIN_SRC}/rpm.bin \
-  flash tz ${VENDOR_PARTITION_BIN_SRC}/tz.bin \
-  flash tzbak ${VENDOR_PARTITION_BIN_SRC}/tz.bin \
-  flash hyp ${VENDOR_PARTITION_BIN_SRC}/hyp.bin \
-  flash hypbak ${VENDOR_PARTITION_BIN_SRC}/hyp.bin \
-  flash modemst1 ${VENDOR_PARTITION_BIN_SRC}/modemst1.bin \
-  flash modemst2 ${VENDOR_PARTITION_BIN_SRC}/modemst2.bin \
-  flash DDR ${VENDOR_PARTITION_BIN_SRC}/DDR.bin \
-  flash fsg ${VENDOR_PARTITION_BIN_SRC}/fsg.bin \
-  flash sec ${VENDOR_PARTITION_BIN_SRC}/sec.bin \
-  flash aboot ${VENDOR_PARTITION_BIN_SRC}/aboot.bin \
-  flash abootbak ${VENDOR_PARTITION_BIN_SRC}/aboot.bin \
-  flash misc ${PARTITION_BIN_SRC}/misc.bin \
-  flash boot_a ${ANDROID_PRODUCT_OUT}/boot.img \
-  flash system_a ${ANDROID_PRODUCT_OUT}/system.img \
-  flash boot_b ${ANDROID_PRODUCT_OUT}/boot.img \
-  flash system_b ${ANDROID_PRODUCT_OUT}/system.img \
-  flash userdata ${ANDROID_PRODUCT_OUT}/userdata.img "$@"
+  flash partition ${VENDOR}/partitions/partition-tables.bin \
+  flash modem ${VENDOR}/vendor_partitions/modem.bin \
+  flash fsc ${VENDOR}/vendor_partitions/fsc.bin \
+  flash sbl1 ${VENDOR}/vendor_partitions/sbl1.bin \
+  flash sbl1bak ${VENDOR}/vendor_partitions/sbl1.bin \
+  flash rpm ${VENDOR}/vendor_partitions/rpm.bin \
+  flash rpmbak ${VENDOR}/vendor_partitions/rpm.bin \
+  flash tz ${VENDOR}/vendor_partitions/tz.bin \
+  flash tzbak ${VENDOR}/vendor_partitions/tz.bin \
+  flash hyp ${VENDOR}/vendor_partitions/hyp.bin \
+  flash hypbak ${VENDOR}/vendor_partitions/hyp.bin \
+  flash modemst1 ${VENDOR}/vendor_partitions/modemst1.bin \
+  flash modemst2 ${VENDOR}/vendor_partitions/modemst2.bin \
+  flash DDR ${VENDOR}/vendor_partitions/DDR.bin \
+  flash fsg ${VENDOR}/vendor_partitions/fsg.bin \
+  flash sec ${VENDOR}/vendor_partitions/sec.bin \
+  flash aboot ${VENDOR}/vendor_partitions/aboot.bin \
+  flash abootbak ${VENDOR}/vendor_partitions/aboot.bin \
+  flash misc ${VENDOR}/partitions/misc.bin \
+  flash boot_a ${OS}/boot.img \
+  flash system_a ${OS}/system.img \
+  flash boot_b ${OS}/boot.img \
+  flash system_b ${OS}/system.img \
+  flash userdata ${OS}/userdata.img "$@"