diff --git a/scripts/dist/tensorflow-driver.sh b/scripts/dist/tensorflow-driver.sh
index 324d7acff..5d75143b9 100755
--- a/scripts/dist/tensorflow-driver.sh
+++ b/scripts/dist/tensorflow-driver.sh
@@ -1,7 +1,8 @@
 #!/bin/bash
 
 PATH="/usr/local/sbin:/usr/sbin:/sbin:/usr/local/bin:/usr/bin:/bin:/scripts"
-CPU_DETECTED=$(/usr/bin/lshw -c processor -json 2>/dev/null)
+
+CPU_DETECTED=$(lshw -c processor -json 2>/dev/null)
 
 if [[ $(echo "${CPU_DETECTED}" | jq -r '.[].capabilities.avx2') == "true" ]]; then
   echo "avx2"