diff --git a/classify_videos.sh b/classify_videos.sh
index 0c88013..8cb9407 100755
--- a/classify_videos.sh
+++ b/classify_videos.sh
@@ -1,4 +1,11 @@
 #!/bin/bash
+if [ -z "${HOME}" ]; then
+    if [ "$(whoami)" == "root" ]; then
+        HOME="/root"
+    else
+        HOME=$(grep "$(whoami)" /etc/passwd | awk 'BEGIN {FS=":"} {print($6)}')
+    fi
+fi
 download_dir="$(grep download_dir "${CONFIG_FILE}" | awk 'BEGIN {FS="="} {print($2)}' | sed 's/"//g')"
 if [ "${download_dir}" == "" ]; then
     download_dir="${HOME}/downloaded_youtube_videos"