I /wish/ the shot enemy would approach your general direction. This triggers combat and they have a bonus to Initiative. The rest of their group may be surprised.
But, I do think after attacking an enemy from stealth, non-rogues should still be able to hide from the alarmed enemy using a bonus action.
They just shouldn't have advantage, nor improved hit rate... /unless/ they move a significant distance away from their og hiding spot. Like 15m away. Enough to waste a turn to regain advantage.
I'm all for stealth tactics and playing cat and mouse, but we need /something/ to... raise opponents' cunning, really 😮💨