Partial return keyword returns "wrong type" for Page when inside `with` clause

Super excited about .55!

In this sense I’m using the new return keyword in a partial to get a page resources and return it.
This works very nicely unless I wrap the return keyword in a with clause. For exemple the following:

{{ with $bundle.Resources.GetMatch $path }}
  {{ return . }}
{{ end }}

Will return the following:

[]interface {}{(*hugolib.pageState)(0xc0009638c0), "en/employees/test.md"}

Meaning I cannot call .Title without:

can't evaluate field Title in type []interface {}