PDIO-342 Use the same prediction key