diff --git a/lib/reducers/device-token-reducer.js b/lib/reducers/device-token-reducer.js
deleted file mode 100644
index c6078ff0e..000000000
--- a/lib/reducers/device-token-reducer.js
+++ /dev/null
@@ -1,26 +0,0 @@
-// @flow
-
-import { setDeviceTokenActionTypes } from '../actions/device-actions.js';
-import type { BaseAction } from '../types/redux-types';
-import { incrementalStateSyncActionType } from '../types/socket-types.js';
-import { updateTypes } from '../types/update-types-enum.js';
-import { ashoatKeyserverID } from '../utils/validation-utils.js';
-
-function reduceDeviceToken(state: ?string, action: BaseAction): ?string {
-  if (action.type === setDeviceTokenActionTypes.success) {
-    return action.payload.deviceTokens[ashoatKeyserverID];
-  }
-  if (action.type === incrementalStateSyncActionType) {
-    for (const update of action.payload.updatesResult.newUpdates) {
-      if (
-        update.type === updateTypes.BAD_DEVICE_TOKEN &&
-        update.deviceToken === state
-      ) {
-        return null;
-      }
-    }
-  }
-  return state;
-}
-
-export { reduceDeviceToken };
diff --git a/lib/reducers/keyserver-reducer.js b/lib/reducers/keyserver-reducer.js
index 24706466b..b0d8b06ae 100644
--- a/lib/reducers/keyserver-reducer.js
+++ b/lib/reducers/keyserver-reducer.js
@@ -1,357 +1,367 @@
 // @flow
 
-import { reduceDeviceToken } from './device-token-reducer.js';
 import { unsupervisedBackgroundActionType } from './lifecycle-state-reducer.js';
 import { updateActivityActionTypes } from '../actions/activity-actions.js';
-import { updateLastCommunicatedPlatformDetailsActionType } from '../actions/device-actions.js';
+import {
+  updateLastCommunicatedPlatformDetailsActionType,
+  setDeviceTokenActionTypes,
+} from '../actions/device-actions.js';
 import {
   addKeyserverActionType,
   removeKeyserverActionType,
 } from '../actions/keyserver-actions.js';
 import { siweAuthActionTypes } from '../actions/siwe-actions.js';
 import {
   logOutActionTypes,
   deleteAccountActionTypes,
   registerActionTypes,
   logInActionTypes,
   resetUserStateActionType,
 } from '../actions/user-actions.js';
 import { queueActivityUpdatesActionType } from '../types/activity-types.js';
-import type { KeyserverStore } from '../types/keyserver-types';
+import type { KeyserverInfos, KeyserverStore } from '../types/keyserver-types';
 import type { BaseAction } from '../types/redux-types.js';
 import {
   fullStateSyncActionType,
   incrementalStateSyncActionType,
   updateConnectionStatusActionType,
   setLateResponseActionType,
   updateDisconnectedBarActionType,
 } from '../types/socket-types.js';
+import { updateTypes } from '../types/update-types-enum.js';
 import { processUpdatesActionType } from '../types/update-types.js';
 import { setNewSessionActionType } from '../utils/action-utils.js';
 import { getConfig } from '../utils/config.js';
 import { setURLPrefix } from '../utils/url-utils.js';
 import { ashoatKeyserverID } from '../utils/validation-utils.js';
 
 export default function reduceKeyserverStore(
   state: KeyserverStore,
   action: BaseAction,
 ): KeyserverStore {
   if (action.type === addKeyserverActionType) {
     return {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [action.payload.keyserverAdminUserID]: {
           ...action.payload.newKeyserverInfo,
         },
       },
     };
   } else if (action.type === removeKeyserverActionType) {
     const { [action.payload.keyserverAdminUserID]: _, ...rest } =
       state.keyserverInfos;
 
     return {
       ...state,
       keyserverInfos: rest,
     };
   } else if (action.type === resetUserStateActionType) {
     // this action is only dispatched on native
     const keyserverInfos = { ...state.keyserverInfos };
     for (const keyserverID in keyserverInfos) {
       const stateCookie = state.keyserverInfos[keyserverID]?.cookie;
       const cookie =
         stateCookie && stateCookie.startsWith('anonymous=')
           ? stateCookie
           : null;
       keyserverInfos[keyserverID] = { ...keyserverInfos[keyserverID], cookie };
     }
 
     state = {
       ...state,
       keyserverInfos,
     };
   } else if (action.type === setNewSessionActionType) {
     const { keyserverID } = action.payload;
     if (action.payload.sessionChange.cookie !== undefined) {
       state = {
         ...state,
         keyserverInfos: {
           ...state.keyserverInfos,
           [keyserverID]: {
             ...state.keyserverInfos[keyserverID],
             cookie: action.payload.sessionChange.cookie,
           },
         },
       };
     }
     if (action.payload.sessionChange.cookieInvalidated) {
       state = {
         ...state,
         keyserverInfos: {
           ...state.keyserverInfos,
           [keyserverID]: {
             ...state.keyserverInfos[keyserverID],
             connection: {
               ...state.keyserverInfos[keyserverID].connection,
               queuedActivityUpdates: [],
             },
           },
         },
       };
     }
   } else if (
     action.type === logInActionTypes.success ||
     action.type === siweAuthActionTypes.success
   ) {
     const { updatesCurrentAsOf } = action.payload;
     for (const keyserverID in updatesCurrentAsOf) {
       state = {
         ...state,
         keyserverInfos: {
           ...state.keyserverInfos,
           [keyserverID]: {
             ...state.keyserverInfos[keyserverID],
             updatesCurrentAsOf: updatesCurrentAsOf[keyserverID],
             lastCommunicatedPlatformDetails: getConfig().platformDetails,
           },
         },
       };
     }
   } else if (action.type === fullStateSyncActionType) {
     const { keyserverID } = action.payload;
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           updatesCurrentAsOf: action.payload.updatesCurrentAsOf,
         },
       },
     };
   } else if (action.type === incrementalStateSyncActionType) {
     const { keyserverID } = action.payload;
+    let { deviceToken } = state.keyserverInfos[keyserverID];
+    for (const update of action.payload.updatesResult.newUpdates) {
+      if (
+        update.type === updateTypes.BAD_DEVICE_TOKEN &&
+        update.deviceToken === state.keyserverInfos[keyserverID].deviceToken
+      ) {
+        deviceToken = null;
+        break;
+      }
+    }
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           updatesCurrentAsOf: action.payload.updatesResult.currentAsOf,
+          deviceToken,
         },
       },
     };
   } else if (action.type === processUpdatesActionType) {
     const { keyserverID } = action.payload;
     const updatesCurrentAsOf = Math.max(
       action.payload.updatesResult.currentAsOf,
       state.keyserverInfos[keyserverID].updatesCurrentAsOf,
     );
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           updatesCurrentAsOf,
         },
       },
     };
   } else if (action.type === setURLPrefix) {
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [ashoatKeyserverID]: {
           ...state.keyserverInfos[ashoatKeyserverID],
           urlPrefix: action.payload,
         },
       },
     };
   } else if (action.type === updateLastCommunicatedPlatformDetailsActionType) {
     const { keyserverID, platformDetails } = action.payload;
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           lastCommunicatedPlatformDetails: platformDetails,
         },
       },
     };
   } else if (action.type === registerActionTypes.success) {
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [ashoatKeyserverID]: {
           ...state.keyserverInfos[ashoatKeyserverID],
           lastCommunicatedPlatformDetails: getConfig().platformDetails,
         },
       },
     };
   } else if (action.type === updateConnectionStatusActionType) {
     const { keyserverID, status } = action.payload;
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           connection: {
             ...state.keyserverInfos[keyserverID].connection,
             status,
             lateResponses: [],
           },
         },
       },
     };
   } else if (action.type === unsupervisedBackgroundActionType) {
     const { keyserverID } = action.payload;
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           connection: {
             ...state.keyserverInfos[keyserverID].connection,
             status: 'disconnected',
             lateResponses: [],
           },
         },
       },
     };
   } else if (action.type === queueActivityUpdatesActionType) {
     const { activityUpdates, keyserverID } = action.payload;
     const oldConnection = state.keyserverInfos[keyserverID].connection;
     const connection = {
       ...oldConnection,
       queuedActivityUpdates: [
         ...oldConnection.queuedActivityUpdates.filter(existingUpdate => {
           for (const activityUpdate of activityUpdates) {
             if (
               ((existingUpdate.focus && activityUpdate.focus) ||
                 (existingUpdate.focus === false &&
                   activityUpdate.focus !== undefined)) &&
               existingUpdate.threadID === activityUpdate.threadID
             ) {
               return false;
             }
           }
           return true;
         }),
         ...activityUpdates,
       ],
     };
 
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           connection,
         },
       },
     };
   } else if (action.type === updateActivityActionTypes.success) {
     const { activityUpdates } = action.payload;
     let keyserverInfos = { ...state.keyserverInfos };
     for (const keyserverID in activityUpdates) {
       const oldConnection = keyserverInfos[keyserverID].connection;
       const queuedActivityUpdates = oldConnection.queuedActivityUpdates.filter(
         activityUpdate =>
           !activityUpdates[keyserverID].includes(activityUpdate),
       );
 
       keyserverInfos = {
         ...keyserverInfos,
         [keyserverID]: {
           ...keyserverInfos[keyserverID],
           connection: { ...oldConnection, queuedActivityUpdates },
         },
       };
     }
     state = {
       ...state,
       keyserverInfos,
     };
   } else if (
     action.type === logOutActionTypes.success ||
     action.type === deleteAccountActionTypes.success
   ) {
     // We want to remove all keyservers but Ashoat's keyserver
     const oldConnection = state.keyserverInfos[ashoatKeyserverID].connection;
 
     const keyserverInfos = {
       [ashoatKeyserverID]: {
         ...state.keyserverInfos[ashoatKeyserverID],
         connection: { ...oldConnection, queuedActivityUpdates: [] },
       },
     };
 
     state = {
       ...state,
       keyserverInfos,
     };
   } else if (action.type === setLateResponseActionType) {
     const { messageID, isLate, keyserverID } = action.payload;
     const lateResponsesSet = new Set<number>(
       state.keyserverInfos[keyserverID].connection.lateResponses,
     );
     if (isLate) {
       lateResponsesSet.add(messageID);
     } else {
       lateResponsesSet.delete(messageID);
     }
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           connection: {
             ...state.keyserverInfos[keyserverID].connection,
             lateResponses: [...lateResponsesSet],
           },
         },
       },
     };
   } else if (action.type === updateDisconnectedBarActionType) {
     const { keyserverID } = action.payload;
     state = {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           connection: {
             ...state.keyserverInfos[keyserverID].connection,
             showDisconnectedBar: action.payload.visible,
           },
         },
       },
     };
-  }
-
-  const deviceToken = reduceDeviceToken(
-    state.keyserverInfos[ashoatKeyserverID].deviceToken,
-    action,
-  );
-  if (deviceToken !== state.keyserverInfos[ashoatKeyserverID].deviceToken) {
+  } else if (action.type === setDeviceTokenActionTypes.success) {
+    const { deviceTokens } = action.payload;
+    const keyserverInfos: { ...KeyserverInfos } = { ...state.keyserverInfos };
+    for (const keyserverID in deviceTokens) {
+      keyserverInfos[keyserverID] = {
+        ...keyserverInfos[keyserverID],
+        deviceToken: deviceTokens[keyserverID],
+      };
+    }
     state = {
       ...state,
-      keyserverInfos: {
-        ...state.keyserverInfos,
-        [ashoatKeyserverID]: {
-          ...state.keyserverInfos[ashoatKeyserverID],
-          deviceToken,
-        },
-      },
+      keyserverInfos,
     };
   }
 
   return state;
 }