Fixed return issue.