Aah, I found that out too after buying the TA05 shafts. I then had to buy the WR-02 shafts too:
The #53792 (tt01) & #54394 (wr02) are the same length but the #54394 (wr02) come with the longer stub axles. These are both for the shorter WR-02 arms. If you want universals for the comical length arms these are #53847 (wr02cb). However, these don't come with the longer stub axles. On my comical I wanted it to be a little wider so I used the #54394 (wr02cb) shafts with the #53847 (wr02) stub axles. This allowed me to then use the longer 9mm hexes. (#54610)
Here's the option parts list for both the WR-02 and WR-02CB
https://d7z22c0gz59ng.cloudfront.net/cms/img/usr/inst/parts/wr02_PARTS_CATALOG.pdf
https://d7z22c0gz59ng.cloudfront.net/cms/img/usr/inst/parts/wr02cb_PARTS_CATALOG.pdf
Hope this helps.