diff --git a/lib/reducers/keyserver-reducer.js b/lib/reducers/keyserver-reducer.js
index b0d8b06ae..ce87afd3c 100644
--- a/lib/reducers/keyserver-reducer.js
+++ b/lib/reducers/keyserver-reducer.js
@@ -1,367 +1,370 @@
 // @flow
 
 import { unsupervisedBackgroundActionType } from './lifecycle-state-reducer.js';
 import { updateActivityActionTypes } from '../actions/activity-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 { 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 = {
+    return {
       ...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: [],
             },
           },
         },
       };
     }
+    return state;
   } else if (
     action.type === logInActionTypes.success ||
     action.type === siweAuthActionTypes.success
   ) {
     const { updatesCurrentAsOf } = action.payload;
+    let keyserverInfos = { ...state.keyserverInfos };
     for (const keyserverID in updatesCurrentAsOf) {
-      state = {
-        ...state,
-        keyserverInfos: {
-          ...state.keyserverInfos,
-          [keyserverID]: {
-            ...state.keyserverInfos[keyserverID],
-            updatesCurrentAsOf: updatesCurrentAsOf[keyserverID],
-            lastCommunicatedPlatformDetails: getConfig().platformDetails,
-          },
+      keyserverInfos = {
+        ...keyserverInfos,
+        [keyserverID]: {
+          ...keyserverInfos[keyserverID],
+          updatesCurrentAsOf: updatesCurrentAsOf[keyserverID],
+          lastCommunicatedPlatformDetails: getConfig().platformDetails,
         },
       };
     }
+    return {
+      ...state,
+      keyserverInfos,
+    };
   } else if (action.type === fullStateSyncActionType) {
     const { keyserverID } = action.payload;
-    state = {
+    return {
       ...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 = {
+    return {
       ...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 = {
+    return {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           updatesCurrentAsOf,
         },
       },
     };
   } else if (action.type === setURLPrefix) {
-    state = {
+    return {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [ashoatKeyserverID]: {
           ...state.keyserverInfos[ashoatKeyserverID],
           urlPrefix: action.payload,
         },
       },
     };
   } else if (action.type === updateLastCommunicatedPlatformDetailsActionType) {
     const { keyserverID, platformDetails } = action.payload;
-    state = {
+    return {
       ...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 = {
+    return {
       ...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 = {
+    return {
       ...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 = {
+    return {
       ...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 = {
+    return {
       ...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 = {
+    return {
       ...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 = {
+    return {
       ...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 = {
+    return {
       ...state,
       keyserverInfos: {
         ...state.keyserverInfos,
         [keyserverID]: {
           ...state.keyserverInfos[keyserverID],
           connection: {
             ...state.keyserverInfos[keyserverID].connection,
             showDisconnectedBar: action.payload.visible,
           },
         },
       },
     };
   } 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 = {
+    return {
       ...state,
       keyserverInfos,
     };
   }
 
   return state;
 }