Diogo Jota comes to the rescue