Fix a wrong return value type