ml: fix python test