1 files changed, 9 insertions, 7 deletions
diff --git a/examples/networkdisconnect.rs b/examples/networkdisconnect.rs
index b3e0721..9676dc9 100644
--- a/examples/networkdisconnect.rs
+++ b/examples/networkdisconnect.rs
@@ -6,11 +6,13 @@ use std::env;
 fn main() {
     let docker = Docker::new();
     let networks = docker.networks();
-    let container_id = env::args().nth(1).unwrap();
-    let network_id = env::args().nth(2).unwrap();
-    let info = networks
-        .get(&network_id)
-        .disconnect(&ContainerConnectionOptions::new(&container_id))
-        .unwrap();
-    println!("{:?}", info);
+    match (env::args().nth(1), env::args().nth(2)) {
+        (Some(container_id), Some(network_id)) => println!(
+            "{:?}",
+            networks
+                .get(&network_id)
+                .disconnect(&ContainerConnectionOptions::new(&container_id))
+        ),
+        _ => eprintln!("please provide a container_id and network_id"),
+    }
 }