fixed issue in predictions