modify 1 argument of Motion nbt (y velocity)

I want to modify the data of an entity so only it's y Motion changes, so it's x and y speeds wont be set to 0. I tried execute as @e[tag=ai] run data modify entity @s Motion[0] set value 2, but this doesnt work