Nel 2016 rimane nello stesso team, ottenendo come miglior risultato un quarto posto in Olanda e terminando la stagione all'8º posto con 134 punti. Per la stagione 2017 lascia il team Tech 3 per accasarsi alla squadra ufficiale KTM, dove ritrova il compagno di squadra delle stagioni precedenti, ossia il britannico Bradley Smith. Ottiene come miglior risultato un nono posto in Repubblica Ceca e termina la stagione al 17º posto con 55 punti. Nel 2018 rimane nello stesso team, ottenendo un terzo posto in Comunità Valenciana e terminando la stagione al 14º posto con 51 punti. In questa stagione è costretto a saltare i Gran Premi di Repubblica Ceca, Austria e Gran Bretagna per la frattura della clavicola sinistra rimediata nel warm up del GP della Repubblica Ceca e il Gran Premio d'Aragona per una nuova frattura dello stesso osso rimediata nelle prove libere del GP. In occasione del round inglese, il suo posto in squadra viene preso dal francese Loris Baz. Nel Gran Premio della Comunità Valenciana conquista il suo primo podio in MotoGP, un terzo posto. Chiude la stagione al quattordicesimo posto in classifica piloti. Nel 2019 rimane nello stesso team, con compagno di squadra Johann Zarco; ottiene come miglior risultato un sesto posto in Francia e termina la stagione all'11º posto con 100 punti. In questa stagione è costretto a saltare il Gran Premio d'Aragona a causa della frattura del radio sinistro rimediata nelle qualifiche del GP.