Foliage normals not working as expected

Hey! I am making a stylized grass shader that should (with lack for a better word) “clump” the grass together into one single color. To do this, I align all normals in the same direction. This however comes with a problem that the ground underneath is a slight different color due to normals not aligning. A solution should be to use the “align to surface” mode on my grass type and then getting the up vector in local space for my normals. This however does not work on specifically my grass.

This is how the grass normals look. Notice how the ground underneath has other normals.

Here is the only code I have for the normals (with use tangent space normals set to false):

This exact material functions perfectly if I place the grass mesh (or any other mesh) by hand and rotate it. It just doesn’t work with the landscape grass type. Is this a bug? Or some hidden flag I have missed?

This is how it looks on flat ground which is how I want it everywhere.

And this is how it looks on a slope when the normals get misaligned.