On 22 February 2017, Bálor returned to NXT to assist Shinsuke Nakamura, who was outnumbered by Andrade "Cien" Almas and Bobby Roode. On 10 March, Bálor made his in–ring return at a live event in a six–man tag team match, teaming with Chris Jericho and Sami Zayn to defeat Kevin Owens, Samoa Joe and Triple H. Bálor made his television return after WrestleMania 33 on 3 April episode of "Raw", where he teamed with his former rival Seth Rollins as the two defeated Owens and Joe in a tag team match. Over the following weeks, Bálor would score victories over Jinder Mahal and Curt Hawkins while also receiving an ominous warning from Bray Wyatt and confronting Luke Gallows and Karl Anderson. On 4 June at Extreme Rules, Bálor competed in a fatal five–way extreme rules match to determine the number one contender for Brock Lesnar's Universal Championship, in which Bálor lost after he passed out from Samoa Joe's "Coquina Clutch".