small fix

oxidecomputer · Aug 18, 2023 · 6d86d77 · 6d86d77
1 parent 9790607
commit 6d86d77
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/gateway-sp-comms/src/single_sp.rs b/gateway-sp-comms/src/single_sp.rs
@@ -473,12 +473,10 @@ impl SingleSp {
         } else {
             MgsRequest::ComponentSetActiveSlot { component, slot }
         };
-        self.rpc(msg).await.and_then(|(_peer, response, _data)| {
-            if persist {
-                expect_response!(response, ComponentSetAndPersistActiveSlotAck)
-            } else {
-                expect_response!(response, ComponentSetActiveSlotAck)
-            }
+        self.rpc(msg).await.and_then(if persist {
+            expect_fn!(ComponentSetAndPersistActiveSlotAck)
+        } else {
+            expect_fn!(ComponentSetActiveSlotAck)
         })
     }