diff --git a/requirements.txt b/requirements.txt
index c84bce0..074426f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-tflite_runtime
+tensorflow
 librosa
 pytz
 tzlocal
diff --git a/scripts/install_helpers.sh b/scripts/install_helpers.sh
index 75fd013..22f0d03 100644
--- a/scripts/install_helpers.sh
+++ b/scripts/install_helpers.sh
@@ -38,7 +38,7 @@ get_tf_whl () {
   if [ -n "$WHL" ]; then
     {
       curl -L -o $HOME/BirdNET-Pi/$WHL $BASE_URL$WHL
-      sed "s/tflite_runtime.*/$WHL/" $HOME/BirdNET-Pi/requirements.txt > requirements_custom.txt
+      sed "s/tensorflow.*/$WHL/" $HOME/BirdNET-Pi/requirements.txt > requirements_custom.txt
     }
   fi
 }