diff --git a/native/account/logged-out-modal.react.js b/native/account/logged-out-modal.react.js
--- a/native/account/logged-out-modal.react.js
+++ b/native/account/logged-out-modal.react.js
@@ -291,9 +291,9 @@
         },
       }),
     );
-    const setRegisterState = setStateForContainer(
+    const setRegisterState = setStateForContainer<State, RegisterState>(
       this.guardedSetState,
-      (change: $Shape<RegisterState>) => (fullState: State) => ({
+      (change: Partial<RegisterState>) => (fullState: State) => ({
         registerState: {
           ...fullState.registerState,
           state: { ...fullState.registerState.state, ...change },