fix naming in mdls predict