One subtle detail you can add to your auth flow is, pass phone number from login to forgot password page to prefill there:
Why? The user typed the phone number and password, navigating to forgot password page afterwards is usually when they hit "Wrong credentials" error. So they are probably there for resetting the password for the same account.
Same for registration: if user navigates to registration page after typing phone number in login page, they are probably thinking "Ahh, I don't have an account here, let me register".