President Trump Donald John TrumpSteele Dossier sub-source was subject of FBI counterintelligence probe Pelosi slams Trump executive order on pre-existing conditions: It 'isn't worth the paper it's signed on' Trump 'no longer angry' at Romney because of Supreme Court stance MORE was all smiles on Wednesday as he welcomed congressional Republicans to the White House to celebrate the shared victory on taxes.

Trump heaped praise on Senate Majority Leader Mitch McConnell Addison (Mitch) Mitchell McConnellTrump 'no longer angry' at Romney because of Supreme Court stance On The Money: Anxious Democrats push for vote on COVID-19 aid | Pelosi, Mnuchin ready to restart talks | Weekly jobless claims increase | Senate treads close to shutdown deadline The Hill's Campaign Report: Trump faces backlash after not committing to peaceful transition of power MORE (R-Ky.), who he harshly criticized during the failed ObamaCare repeal effort, and gave a prominent spot near the podium to Sen. Dean Heller Dean Arthur HellerOn The Trail: Democrats plan to hammer Trump on Social Security, Medicare Lobbying World Democrats spend big to put Senate in play MORE (R-Nev.), who’s considered the most vulnerable GOP senator up for reelection next year.

Trump himself said the passage of the tax bill made it easier for all of the Republicans to be in a good mood.

ADVERTISEMENT

“We got together and we worked very hard, didn't we?” Trump said from the South Lawn of the White House. “Seems like it was a lot of fun. It’s always fun when you win. If you work hard and lose, that's not acceptable.”

But it's unclear whether the Wednesday reverie will carry over into 2018, especially as the party heads for midterm elections where the House and Senate could be at risk.

Trump has clashed with several GOP senators in recent month, including Sens. Jeff Flake Jeffrey (Jeff) Lane FlakeHow fast population growth made Arizona a swing state Jeff Flake: Republicans 'should hold the same position' on SCOTUS vacancy as 2016 Republican former Michigan governor says he's voting for Biden MORE (Ariz.), Bob Corker Robert (Bob) Phillips CorkerHas Congress captured Russia policy? Tennessee primary battle turns nasty for Republicans Cheney clashes with Trump MORE (Tenn.) and John McCain John Sidney McCainAnalysis: Biden victory, Democratic sweep would bring biggest boost to economy The Memo: Trump's strengths complicate election picture Mark Kelly: Arizona Senate race winner should be sworn in 'promptly' MORE (Ariz.). Corker, a Trump surrogate-turned-critic who announced his retirement in September, ended up getting on board with tax legislation in the end. But Flake, who has publicly sparred with Trump and is also retiring from Congress, skipped the White House event despite voting for the bill.

Some vulnerable Republicans may feel pressure to distance themselves from Trump in 2018, given his low approval ratings. But that could be a risky bet, since they will also need the support of the GOP base to survive primary challenges and triumph in November.

That puts Heller in a particularly tricky spot as he looks to navigate a contentious primary and competitive general election in Nevada, a state that seems to be trending toward the Democrats.

When it comes to the primary, Heller’s center-stage spot at the White House is good news for the Nevada Republican as a way to prove to the base that he’ll work with the president on big-ticket legislative items.

ADVERTISEMENT

Heller has had a rocky relationship with Trump that goes back to the 2016 election. He was critical of Trump as a presidential candidate and, during Trump’s first year in office, the two clashed during the unsuccessful ObamaCare repeal push.

At a meeting in the White House, Trump called out Heller — who was sitting next to him — for not getting on board with the initial repeal bill. A pro-Trump outside group threatened to run ads against Heller and later ended up pulling them when the Nevada senator ultimately voted for the legislation.

Heller seemed to have gotten back into Trump’s good graces by pushing a later ObamaCare repeal bill alongside Sens. Lindsey Graham Lindsey Olin GrahamSteele Dossier sub-source was subject of FBI counterintelligence probe Hillicon Valley: Subpoenas for Facebook, Google and Twitter on the cards | Wray rebuffs mail-in voting conspiracies | Reps. raise mass surveillance concerns Key Democrat opposes GOP Section 230 subpoena for Facebook, Twitter, Google MORE (R-S.C.) and Bill Cassidy William (Bill) Morgan CassidyCoushatta tribe begins long road to recovery after Hurricane Laura Senators offer disaster tax relief bill Bottom line MORE (R-La.).

Support for Trump could be critical for Heller’s chances in the GOP primary, where he’ll need to convince voters that he’s sincerely embracing the president. And even if the president doesn’t endorse him, Heller will need to at least make sure that Trump isn’t actively working against him.

Trump’s former chief strategist Stephen Bannon, who’s vowed to go to “war” with McConnell and his allies, is expected to go to bat for Heller’s GOP primary rival Danny Tarkanian, who has unsuccessfully run for office several times and lost a close Nevada House race in 2016.

Tarkanian has played up his support for Trump and has called on Heller to sign a pledge to oppose McConnell remaining as majority leader.

But Heller’s been loudly trumpeting his role in writing and passing the GOP’s tax overhaul. While some strategists believe Heller’s conflict with Trump in 2016 could still hurt him, passage of the tax bill could be a boon for him in the June 12 primary.

“His first objective is to win the primary, so in terms of accomplishing that goal, this is very helpful. It’s getting something done and having a shared accomplishment with the president that he can tout in the primary,” said GOP strategist Brian Seitchik, who worked on Tarkanian’s 2010 Senate bid but isn’t currently working on either campaign.

“It’s hard to really say certainly where the relationship with the president is headed. I would say today it’s probably all smiles and sunshine.”

But if Heller wins the primary, his association with Trump could go from an asset to a liability.

Hillary Clinton Hillary Diane Rodham ClintonFox News poll: Biden ahead of Trump in Nevada, Pennsylvania and Ohio Trump, Biden court Black business owners in final election sprint The power of incumbency: How Trump is using the Oval Office to win reelection MORE won Nevada in 2016, so Heller’s support for Trump could cost him more moderate and independent voters, especially if the president’s approval numbers are still underwater by 2018. He’s also likely to have a tough Democratic opponent, with Rep. Jacky Rosen Jacklyn (Jacky) Sheryl RosenSenators introduce bipartisan bill to help women, minorities get STEM jobs Hillicon Valley: Election officials prepare for new Russian interference battle | 'Markeyverse' of online fans helps take down a Kennedy | GOP senators unveil bill to update tech liability protections Google, Apple, eBay to meet virtually with lawmakers for tech group's annual fly-in MORE (Nev.) favored to win her party’s primary.

Photos of Heller standing directly behind Trump could be a double-edged sword as Democrats continue to link the Nevada Republican to the president in a state where he lost by more than 2 points. And the party plans to weaponize the tax overhaul, as well as ObamaCare repeal, against him.

ADVERTISEMENT

“This tax plan was never about bringing relief to the working families and small businesses who need it most — it was about pandering to Republican primary voters and sucking up to billionaire mega-donors,” Nevada Democratic Party chair William McCurdy II said after the bill passed. “Nevadans will remember Senator Heller’s unwavering support for this shameful tax bill in 2018.”

While it’s still unclear if Trump will be a factor in every 2018 race or continue to mend fences with Senate Republicans, the president and his allies appear to be going to bat for other GOP incumbents who could face tough intraparty battles.

The 45 Committee, a pro-Trump outside group, ran a six-figure ad buy touting Sen. Roger Wicker Roger Frederick WickerHillicon Valley: Subpoenas for Facebook, Google and Twitter on the cards | Wray rebuffs mail-in voting conspiracies | Reps. raise mass surveillance concerns Key Democrat opposes GOP Section 230 subpoena for Facebook, Twitter, Google Senate panel threatens subpoena for Google, Facebook and Twitter executives MORE (R-Miss.) for supporting tax reform. The group was founded by GOP mega-donors Sheldon Adelson and Todd Ricketts to help support the president’s agenda.

Unlike Heller, Wicker is up for reelection in 2018 in a safe red seat, but he could also face a primary challenge from state Sen. Chris McDaniel, who came close to unseating Sen. Thad Cochran William (Thad) Thad CochranEspy wins Mississippi Senate Democratic primary Bottom Line Mike Espy announces Mississippi Senate bid MORE (R-Miss.) in 2014.

Bannon has also been eyeing Mississippi’s primary and pushing McDaniel to run. But Trump has reportedly vowed to back Wicker, and support from his allied groups could convince McDaniel to seek another higher office since he’s also considering a run for lieutenant governor.

Trump’s support for Senate incumbents could put him at odds with Bannon, but help neutralize primary battles that could threaten the GOP’s slim Senate majority. And strategists believe lawmakers who demonstrate that they stand by Trump will reap the benefits of winning over his base.

“I think folks who have a record of standing by the president will be rewarded by the president’s supporters,” Seitchik said.