Fixing that part of stealth would be nice, but I believe the bigger issue by far is Hide being a bonus action for everyone. Make it a full action (except for the Rogue of course) and Hide for advantage is already much less of an imbalance simply due to action economy.
I happen to disagree and think that the bigger issue is precisely the lack of having to pass a stealth check over it being a bonus action.
Why do I say so ? Because it's bad enough to break the game even when you are using a rogue, the class that SHOULD have it a as bonus action to begin with.
Conversely the "bonus action hide" in itself wouldn't be so powerful for other characters if they had to pass a check (maybe even one with disadvantage, depending on their armor/status) to use it effectively.