diff --git a/onvif/src/discovery/mod.rs b/onvif/src/discovery/mod.rs
index 4997993..bbaf0ff 100644
--- a/onvif/src/discovery/mod.rs
+++ b/onvif/src/discovery/mod.rs
@@ -327,13 +327,12 @@ impl DiscoveryBuilder {
         } = self;
 
         match discovery_mode {
-            DiscoveryMode::Multicast => self.run_multicast(duration, listen_address),
+            DiscoveryMode::Multicast => self.run_multicast(duration, listen_address).await,
             DiscoveryMode::Unicast {
                 network,
                 network_mask,
-            } => self.run_unicast(duration, listen_address, network, network_mask),
+            } => self.run_unicast(duration, listen_address, network, network_mask).await,
         }
-        .await
     }
 }