Fix a wrong return type