ugtrain: fix segfault in disc stage != 4