Fixed a missing 'return' issue.

BTW, all tests passed.