Beginning in 2004 , the winner had their choice of a World Heavyweight Championship or WWE Championship match at WrestleMania . Despite being introduced in late 2002 , it was not until 2004 that a winner had the option to choose a match for the World Heavyweight Championship instead of the WWE Championship at WrestleMania .