Fixed a missing 'return' issue.