If they don't stick with Trump they would have to worry about a primary challenge

At this point, only Ben Sasse, Mitt Romney and Susan Collins have decided they don't need Trump to keep their seats. Everyone else is too afraid to move on from him.

If Trump stays in the spotlight, I don't see that changing.