Question Flatshaded terrain with different textures per quad depending on height

Hi! I recently started working with the shader graph in Unity, and I would like to achieve a flatshaded effect with different textures depending on the height of the mesh, like at 7:10 in this procedural terrain tutorial:

As in the tutorial, I want the texture to be per quad, but still dependent on height.
Right now, I have different textures based on height, and they have a lerp between them. But I want the cutoff between textures to be sharp and be per quad.

How would I achieve this?

 

make gradient texture, treat the normal as a uv coordinate, and tex2d(i.normal.y). use a flat normal mesh, or try squiggling it round()