WASD Grid Based Movement "Add Button With One Modifier Composite"

Hello,

I have a simple set-up with the new Input System with WASD for grid based movement. User presses “W”, then the player moves 1 grid forward, etc. Every thing works fine with this set-up.

I would like to implement a modifier key to allow the player to move 2 grids. In a perfect world, the user would hold “shift” then press “W” to move 2 steps forward. I’ve tried to set it up like this:

I’m probably missing something, but it isn’t working the way I wish it would. Pressing “W” with or without “shift” make the player move 1 step. I think I’m going the right direction with the set-up but, I see that the Composite part of my WASD for “W” is set to Up:

Meanwhile, the Composite Part for my “W” in the “Extra Step W” Action is set to: Button. Is this what I’m missing for this to work properly? How to I change this to be “Up” instead of “Button”?

I’ve figured out a way to get the expected behavior. But something tells me it’s a bit clumsy. I’ve added two actions: “AddPossibleSteps” with “Shift key” pressed and “RemovePossible” with “Shift key” release. Then added two methods in script where each method respectively adds or remove a “integer possibleSteps”.

Is there a way to make this work by using “Add Button With One Modifier Composite”?