A cooldown is not really a fix. Part of the issue is pretty much everyone can Bonus Action hide when only rogues are supposed to be allowed to do that, which actually makes it that ever class can just attack then hide, and a cooldown would just make it that they wander out of the situation until the cooldown is done and the repeat the process.
