Fixed incorrect return type
Fixed incorrect return type #4