diff --git a/common/init b/common/init
index f68cbfc..a69c615 100644
--- a/common/init
+++ b/common/init
@@ -4,11 +4,11 @@
 
 needs_update=true
 
-. ~/.last_revision 2>/dev/null || true
+. $SNAP_USER_DATA/.last_revision 2>/dev/null || true
 if [ "$SNAP_DESKTOP_LAST_REVISION" = "$SNAP_REVISION" ]; then
   needs_update=false
 fi
-[ $needs_update = true ] && echo "SNAP_DESKTOP_LAST_REVISION=$SNAP_REVISION" > ~/.last_revision
+[ $needs_update = true ] && echo "SNAP_DESKTOP_LAST_REVISION=$SNAP_REVISION" > $SNAP_USER_DATA/.last_revision
 
 if [ "$SNAP_ARCH" == "amd64" ]; then
   ARCH="x86_64-linux-gnu"
diff --git a/common/mark-and-exec b/common/mark-and-exec
index 6f0660d..8636322 100644
--- a/common/mark-and-exec
+++ b/common/mark-and-exec
@@ -2,6 +2,6 @@
 # Mark update and exec binary #
 ###############################
 
-[ $needs_update = true ] && echo "SNAP_DESKTOP_LAST_REVISION=$SNAP_REVISION" > ~/.last_revision
+[ $needs_update = true ] && echo "SNAP_DESKTOP_LAST_REVISION=$SNAP_REVISION" > $SNAP_USER_DATA/.last_revision
 
 exec "$@"