Nell'estate 2016 fa ritorno al Milan, ora guidato da Montella. Il 27 agosto segna la prima rete in rossonero, nella gara persa 4-2 contro il Napoli. Nel derby del 20 novembre realizza una doppietta, trovando al contempo il duecentesimo gol nelle stracittadine dell'era Berlusconi. Il 23 dicembre vince la Supercoppa italiana, con la squadra milanese che batte ai rigori la Juventus, servendo anche un assist a Giacomo Bonaventura.