Material Function Issue

I have a node network I had inside a material working, so decided to turn it into a function for reusability.

However it does not allow me to connect a FunctionInput Vector 3 into the Tex input of my Texture sample,

but I can find examples of this being done in the flowmap material functions,

Any ideas?