Fix response decoding for bulk trait upload (#28)

Flagsmith · Feb 22, 2023 · f57093d · f57093d
1 parent f18d390
commit f57093d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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))
     }
   }