Grenadier notable not working when weapon swapping?

Hi!

I play a Tactician with grenades and ballistas, and it's really fun.

I just started messing with the weapon swap mechanic. So I have all my grenade skills on weapon set 1 and my ballistas and explosive shot on weapon swap 2.

I have allocated the Grenadier notable(Grenade Skills have +1 Cooldown Use) on weapon set 1 and +max summons on weapon set 2.

The summons still keep the max three ballistas when i swap to grenade skills but when I use a skill that is set to weapon set 2 then my maximum grenade cooldown use is reset to 3 instead of the four that it have on weapon set 1.

So I have 4 grenades ready then summon a ballista and it resets to 3, and when i use a grenade all of the cooldowns are able to start "loading" up to the forth one again.

Is this intended or a bug?

If anyone have ideas please replay to this thread. Any ideas are welcome.
My tree: https://maxroll.gg/poe2/planner/m63z60lb