diff --git a/instagrapi/mixins/challenge.py b/instagrapi/mixins/challenge.py
index 0f29976c..550c5be0 100644
--- a/instagrapi/mixins/challenge.py
+++ b/instagrapi/mixins/challenge.py
@@ -186,7 +186,7 @@ def challenge_resolve_contact_form(self, challenge_url: str) -> bool:
         ), result
         for retry_code in range(5):
             for attempt in range(1, 11):
-                code = self.challenge_code_handler(self.username, choice)
+                code = self.challenge_code_handler(self.username, choice, self.code)
                 if code:
                     break
                 time.sleep(WAIT_SECONDS * attempt)
@@ -388,7 +388,7 @@ def challenge_resolve_simple(self, challenge_url: str) -> bool:
                     raise ChallengeError(f'ChallengeResolve: Choice "email" or "phone_number" (sms) not available to this account {self.last_json}')
             wait_seconds = 5
             for attempt in range(24):
-                code = self.challenge_code_handler(self.username, ChallengeChoice.EMAIL)
+                code = self.challenge_code_handler(self.username, ChallengeChoice.EMAIL, self.code)
                 if code:
                     break
                 time.sleep(wait_seconds)