diff --git a/install.sh b/install.sh
index 6407a58..9f621be 100644
--- a/install.sh
+++ b/install.sh
@@ -1,15 +1,33 @@
 #!/bin/sh
 
-user_authorized_keys_file="/mnt/data/ssh/authorized_keys"
+# Get DataDir location
+DATA_DIR="/mnt/data"
+case "$(ubnt-device-info firmware || true)" in
+    1*)
+      DATA_DIR="/mnt/data"
+      ;;
+    2*)
+      DATA_DIR="/data"
+      ;;
+    3*)
+      DATA_DIR="/data"
+      ;;
+    *)
+      echo "ERROR: No persistent storage found." 1>&2
+      exit 1
+      ;;
+  esac
+
+user_authorized_keys_file="$DATA_DIR/ssh/authorized_keys"
 
 curl -sO https://raw.githubusercontent.com/fire1ce/UDM-Persistent-SSH-Keys/main/99-ssh-keys.sh
-mv 99-ssh-keys.sh /mnt/data/on_boot.d/99-ssh-keys.sh
-chmod +x /mnt/data/on_boot.d/99-ssh-keys.sh
+mv 99-ssh-keys.sh $DATA_DIR/on_boot.d/99-ssh-keys.sh
+chmod +x $DATA_DIR/on_boot.d/99-ssh-keys.sh
 echo "99-ssh-keys.sh installed"
 
 if [ ! -f "$user_authorized_keys_file" ]; then
   echo "Creating $user_authorized_keys_file"
-  mkdir -p /mnt/data/ssh
+  mkdir -p $DATA_DIR/ssh
   touch "$user_authorized_keys_file"
   chmod 0644 "$user_authorized_keys_file"
 fi