Nell'estate 2016 torna all'Atalanta, facendo il suo secondo esordio in campionato con la maglia bergamasca il 2 ottobre 2016, alla 7ª giornata, nella partita casalinga vinta 1-0 contro il . Il successivo 26 ottobre realizza il suo primo gol in massima categoria, decisivo per la vittoria 1-0 in trasferta contro il . Le prestazioni offerte attirano l'interesse della , che il 12 gennaio 2017 ne acquista il cartellino per 15 milioni di euro; contestualmente i bianconeri lo lasciano in prestito a Bergamo fino al 30 giugno 2018. Chiude la stagione in campionato con 30 presenze e 7 reti, tra cui la prima doppietta in carriera, messa a segno il 25 febbraio sul campo del Napoli.