fixing up pred array