Fixed return type mismatch issue.