Košťálek pelasi alle 20-vuotiaiden MM-kilpailuissa vuosina 2014 ja 2015, joista jälkimmäisen turnauksen jäähypörssin hän voitti 34 minuutillaan. Košťálek jakoi myös Tšekin sisäisen tehotilaston voiton David Kašen kanssa lukemalla +3. Hänet valittiin Tanskaa vastaan pelatun alkulohkon ottelun päätteeksi joukkueensa ottelun parhaaksi pelaajaksi.