From f57093d9d5839827cbc1f631bca5e95b791c1724 Mon Sep 17 00:00:00 2001
From: Christian Treffs <ctreffs@gmail.com>
Date: Wed, 22 Feb 2023 11:54:43 +0100
Subject: [PATCH] Fix response decoding for bulk trait upload (#28)

---
 FlagsmithClient/Classes/Flagsmith.swift | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/FlagsmithClient/Classes/Flagsmith.swift b/FlagsmithClient/Classes/Flagsmith.swift
index 7ea947f..c8565dc 100644
--- a/FlagsmithClient/Classes/Flagsmith.swift
+++ b/FlagsmithClient/Classes/Flagsmith.swift
@@ -199,8 +199,8 @@ public class Flagsmith {
   public func setTraits(_ traits: [Trait],
                         forIdentity identity: String,
                         completion: @escaping (Result<[Trait], Error>) -> Void) {
-    apiManager.request(.postTraits(identity: identity, traits: traits)) { (result: Result<[Trait], Error>) in
-      completion(result)
+    apiManager.request(.postTraits(identity: identity, traits: traits)) { (result: Result<Traits, Error>) in
+        completion(result.map(\.traits))
     }
   }