Fixed a return type issue.