I actually found the upgrade, it does feel much better with it, but I still managed to overheat on last bossfight somehow, just for the record