If you begin running out of energy units, allow yourself to die. The moment you are brought back to life, you will get an energy unit for free. For example, if you let yourself die five times in a row you will get five energy units for free. Do this trick each time you begin a new chapter.