Fixed incorrect return type in fluent API