Gustafsson pelasi myös alle 18-vuotiaiden MM-kilpailuissa vuosina 2017 ja 2018, joista jälkimmäinen turnaus päättyi pronssiin. Hän jakoi viidellä syöttöpisteellään Ruotsin sisäisen syöttöpörssin voiton Jonatan Berggrenin kanssa. Gustafssonilla oli myös turnauksen paras aloitusprosentti (70,55%). Valmennus valitsi hänet joukkueensa turnauksen yhdeksi kolmesta parhaasta pelaajasta. Gustafsson valittiin myös Valko-Venäjää vastaan pelatun alkulohkon ottelun päätteeksi Ruotsin ottelun parhaaksi pelaajaksi.