diff --git a/javascript/src/iframe/viamapi-iframe.js b/javascript/src/iframe/viamapi-iframe.js
index 49a8c98f567fef676cbfcf482453e5335b9f2ba0..5e19250878700151465ac3fbd359f15cdf7b22f1 100644
--- a/javascript/src/iframe/viamapi-iframe.js
+++ b/javascript/src/iframe/viamapi-iframe.js
@@ -1760,7 +1760,7 @@ const connection = Penpal.connectToParent({
           handleIdentityLogin(loginIdentity, data.Uuid, data.Session);
           await getProfileData(loginIdentity);
 
-          if (mode === LOGIN_MODES.PREVIOUSLY_ADDED_DEVICE) {
+          if (mode === LOGIN_MODES.SMS) {
             await setIdentityInLocalStorage(loginIdentity);
           }
         } else if (mode === LOGIN_MODES.NEW_DEVICE) {