Dembele returns to training with the rest of the group