loc: fix for V_max from train