Virth Balázs a 2021-es tokiói olimpia után lett Milák Kristóf edzője. Kettejük kapcsolata kezdetben jól alakult, azonban az elmúlt évben már nem tudta kezelni az úszózsenit, a sajtóban pedig többször is nyíltan kritizálta Milákot és a munkához való hozzáállását. Virth Balázs a Nemzeti Sportnak adott interjúban elmondta, hogy a jövőben már nem ő lesz a kétszeres olimpiai Már nem Virth Balázs Milák Kristóf edzője