Fixed issue referencing function body, not function return.