Avoid the local and return the function directly

via @soveran