I don't think the limitation is bad in general. But there could be a few other approaches you could use or things you could tweak
For example:
- Make the booster energy reset after death
- Make the booster energy recharge as fast as you use it up (or a bit slower so it can't be exploited)
For future games just think of it like "what is the limitation achieving?" and if it is possibly frustrating, think about how to make it less frustrating :)
That's how I do it at least :D