fix model mapping in predict