Ao colocar flexbox no .contato, os elementos ficaram um ao lado do outro

Após o media query, ao tentar reproduzir o figma, a parte de contatos não está igual.

Segue print para entendimento:

Como eu posso, no flexbox, instruir a coluna do meio a descer? Ou seja, formar uma segunda linha?

segue o código especifico do tema ( não consigo colocar o codigo todo por causa da quantidade de caracteres):

CSS

@media screen and (min-width: 1024px){
    .contato__titulo{
        font-size: 24px;
    }

    .contato__email{
        width: 30%;
    }

    .contato{
        display: flex;
        justify-content: center;
        align-items: center;
    }
}

HTML

<section class="contato">
            <div class="contato__descricao">
            <h2 class="contato__titulo">Fique por dentro das novidades!</h2>
            <p class="contato__texto">Atualizações de e-books, novos livros, promoções e outros.</p>
            </div>
            <input type="email" placeholder="Cadastre seu e-mail" class="contato__email">

    </section>

Obrigado!