Fix output from predict