World Position Offset Z only ?

Development Rendering
1

I’ve made a plane for water, using World Position Offset, how can I make it Z only ?

2

Multiply it on vertex normal or mask Z component with ComponentMask finction and append to another vector to get (0,0,Z)