Jota back in training