Yes, I used to mark the driveshaft pieces before taking it out in case this happened. Only recently I learned about the marks. There will be one raised mark on one half and two raised marks on the other half.

Align the marks, relax and be happy.

Good luck,