diff --git a/ts/components/inputs/SessionInput.tsx b/ts/components/inputs/SessionInput.tsx
index 139fc6eb1..eb58a9fe9 100644
--- a/ts/components/inputs/SessionInput.tsx
+++ b/ts/components/inputs/SessionInput.tsx
@@ -90,6 +90,7 @@ const ShowHideButton = (props: { forceShow: boolean; toggleForceShow: Noop; erro
           transform: 'translateY(-50%)',
           ...position,
         }}
+        dataTestId="reveal-recovery-phrase-toggle"
       />
     );
   }
@@ -101,6 +102,7 @@ const ShowHideButton = (props: { forceShow: boolean; toggleForceShow: Noop; erro
       iconSize="medium"
       onClick={props.toggleForceShow}
       style={{ position: 'absolute', top: '50%', transform: 'translateY(-50%)', ...position }}
+      dataTestId="reveal-recovery-phrase-toggle"
     />
   );
 };
diff --git a/ts/components/loading/bar/SessionProgressBar.tsx b/ts/components/loading/bar/SessionProgressBar.tsx
index af5b9d0f5..a3dd274a7 100644
--- a/ts/components/loading/bar/SessionProgressBar.tsx
+++ b/ts/components/loading/bar/SessionProgressBar.tsx
@@ -68,7 +68,7 @@ export function SessionProgressBar(props: Props) {
         {showPercentage ? <StyledText>{Math.floor(progress)}%</StyledText> : null}
       </Flex>
       {subtitle || showPercentage ? <SpacerXL /> : null}
-      <ProgressContainer color={backgroundColor} style={{ width }}>
+      <ProgressContainer color={backgroundColor} style={{ width }} data-testid="loading-animation">
         <Progress
           color={color}
           initial={{ width: 0 }}
diff --git a/ts/components/registration/stages/RestoreAccount.tsx b/ts/components/registration/stages/RestoreAccount.tsx
index 8bf013d1e..d01dc5ec1 100644
--- a/ts/components/registration/stages/RestoreAccount.tsx
+++ b/ts/components/registration/stages/RestoreAccount.tsx
@@ -247,7 +247,7 @@ export const RestoreAccount = () => {
                   onClick={recoverAndFetchDisplayName}
                   text={window.i18n('continue')}
                   disabled={!(!!recoveryPassword && !recoveryPasswordError)}
-                  dataTestId="continue-session-button"
+                  dataTestId="continue-button"
                 />
               </>
             ) : (
@@ -279,7 +279,7 @@ export const RestoreAccount = () => {
                     !(!!recoveryPassword && !recoveryPasswordError) ||
                     !(!!displayName && !displayNameError)
                   }
-                  dataTestId="continue-session-button"
+                  dataTestId="continue-button"
                 />
               </Flex>
             )}