diff --git a/src/main.rs b/src/main.rs
index 0360d2c..b73b724 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -109,7 +109,7 @@ fn main() -> Result<()> {
 
     let payload = gnerate_payload(&mut args)?;
 
-    write_payload(&args.transport, &args.device_name, payload)
+    write_payload(&args.transport, Option::from(&args.device_name), payload)
 }
 
 fn list_devices(transport: &TransportProtocol) -> Result<()> {
@@ -236,7 +236,7 @@ fn gnerate_payload(args: &mut Args) -> Result<PayloadBuffer> {
 
 fn write_payload(
     transport: &TransportProtocol,
-    device_name: &Option<String>,
+    device_name: Option<&String>,
     payload: PayloadBuffer,
 ) -> Result<(), anyhow::Error> {
     match transport {
@@ -245,7 +245,7 @@ fn write_payload(
             .enable_all()
             .build()?
             .block_on(async {
-                BleDevice::single(device_name.as_deref())
+                BleDevice::single(device_name.map(|x| x.as_str()))
                     .await?
                     .write(payload)
                     .await