Make it work with 1 input geometry as well!
Make it work with 1 input geometry as well!