diff --git a/99-ssh-keys.sh b/99-ssh-keys.sh
index e503d02..ec7bb01 100755
--- a/99-ssh-keys.sh
+++ b/99-ssh-keys.sh
@@ -1,9 +1,27 @@
 #!/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"
 udm_authorized_keys_file="/root/.ssh/authorized_keys"
 
-# if /mnt/data/ssh/authorized_keys is missing print error message
+# if $DATA_DIR/ssh/authorized_keys is missing print error message
 if [ ! -f "$user_authorized_keys_file" ]; then
   echo "ERROR: $user_authorized_keys_file is missing"
   exit 1