If I use Wrecking Ball (Warhammer) to activate this passive perk “Safety Measures” then swap weapon, the buff is lost. Is this intentional?

If so, why is it intentional? - You’d think the developers would encourage players to swap weapons as it adds more combinations to the ability rotation, making combat more complex.