Nella stagione successiva trova spazio con l'allenatore Vincenzo Montella, che lo impiega con continuità dopo l'infortunio occorso a Montolivo. Il 2 ottobre 2016 realizza il suo primo gol in Serie A nella partita vinta 4-3 contro il Sassuolo. Il 22 ottobre realizza il gol decisivo nella "classica" contro la Juventus, vinta per 1-0 (riuscendo a far trionfare il Milan sui bianconeri dopo quattro anni), contro cui il 23 dicembre si aggiudica, da titolare, anche la Supercoppa italiana, primo trofeo della sua carriera. Il 25 gennaio 2017, ancora in una sfida con i bianconeri, riceve la sua prima espulsione da professionista. Termina la stagione con 25 presenze in campionato.