diff --git a/src/main.js b/src/main.js
index 4285e70b..378fe95c 100644
--- a/src/main.js
+++ b/src/main.js
@@ -10,7 +10,6 @@ import apiModule from './modules/api.js'
 import configModule from './modules/config.js'
 import chatModule from './modules/chat.js'
 import oauthModule from './modules/oauth.js'
-import subscribeModule from './modules/subscribe.js'
 
 import VueTimeago from 'vue-timeago'
 import VueI18n from 'vue-i18n'
@@ -61,8 +60,7 @@ createPersistedState(persistedStateOptions).then((persistedState) => {
       api: apiModule,
       config: configModule,
       chat: chatModule,
-      oauth: oauthModule,
-      subscribe: subscribeModule
+      oauth: oauthModule
     },
     plugins: [persistedState],
     strict: false // Socket modifies itself, let's ignore this for now.
diff --git a/src/modules/subscribe.js b/src/modules/subscribe.js
deleted file mode 100644
index e705904c..00000000
--- a/src/modules/subscribe.js
+++ /dev/null
@@ -1,21 +0,0 @@
-import registerPushNotifications from '../services/push/push.js'
-
-const subscribe = {
-  state: {
-    token: null,
-    vapidPublicKey: null
-  },
-  mutations: {
-    setCurrentUser (state, user) {
-      state.token = user.credentials
-      if (state.token && state.vapidPublicKey) registerPushNotifications(this)
-    },
-    setInstanceOption (state, { name, value }) {
-      if (name !== 'vapidPublicKey') return
-      state.vapidPublicKey = value
-      if (state.token && state.vapidPublicKey) registerPushNotifications(this)
-    }
-  }
-}
-
-export default subscribe
diff --git a/src/modules/users.js b/src/modules/users.js
index 88ec7115..d2c7fdf6 100644
--- a/src/modules/users.js
+++ b/src/modules/users.js
@@ -1,4 +1,5 @@
 import backendInteractorService from '../services/backend_interactor_service/backend_interactor_service.js'
+import registerPushNotifications from '../services/push/push.js'
 import { compact, map, each, merge } from 'lodash'
 import { set } from 'vue'
 
@@ -119,8 +120,10 @@ const users = {
                   })
 
                   // Fetch our friends
-                  store.rootState.api.backendInteractor.fetchFriends({ id: user.id })
+                  store.rootState.api.backendInteractor.fetchFriends({id: user.id})
                     .then((friends) => commit('addNewUsers', friends))
+
+                  registerPushNotifications(store)
                 })
             } else {
               // Authentication failed
diff --git a/src/services/push/push.js b/src/services/push/push.js
index 923c2178..ffab5918 100644
--- a/src/services/push/push.js
+++ b/src/services/push/push.js
@@ -45,17 +45,17 @@ function askPermission () {
 }
 
 function subscribe (registration, store) {
-  if (!store.state.config.webPushNotifications) {
+  if (!store.rootState.config.webPushNotifications) {
     return Promise.reject(new Error('Web Push is disabled in config'))
   }
 
-  if (!store.state.subscribe.vapidPublicKey) {
+  if (!store.rootState.instance.vapidPublicKey) {
     return Promise.reject(new Error('VAPID public key is not found'))
   }
 
   const subscribeOptions = {
     userVisibleOnly: true,
-    applicationServerKey: urlBase64ToUint8Array(store.state.subscribe.vapidPublicKey)
+    applicationServerKey: urlBase64ToUint8Array(store.rootState.instance.vapidPublicKey)
   }
   return registration.pushManager.subscribe(subscribeOptions)
 }
@@ -65,7 +65,7 @@ function sendSubscriptionToBackEnd (subscription, store) {
     method: 'POST',
     headers: {
       'Content-Type': 'application/json',
-      'Authorization': `Bearer ${store.state.subscribe.token}`
+      'Authorization': `Bearer ${store.rootState.oauth.token}`
     },
     body: JSON.stringify({
       subscription,