V ročníku 2020/2021 bol Baláž jeden z kľúčových hráčov Nitry v ceste za postupom do finále, pričom si premiérovo vybojoval reprezentačné pozvánky. Následne zamieril do Karlových Varov, v ktorých už v minulosti pôsobil, no v drese českého extraligistu nemal stabilnú pozíciu a počas sezóny 2022/2023 sa vrátil do Nitry. V jej drese následne odohral 43 zápasov, v ktorých si pripísal 33 kanadských bodov za 18 gólov a 15 asistencií.