final predict fixes