Should Jota have been booked for diving for Liverpool’s overturned penalty?

Should Jota have been booked for diving for Liverpool's overturned penalty?