diff --git a/main.py b/main.py
index 30fc27c..567d6ef 100644
--- a/main.py
+++ b/main.py
@@ -27,7 +27,7 @@ DEVICE_IFACE = 'org.bluez.Device1'
 AGENT_IFACE = 'org.bluez.Agent1'
 AGENT_MANAGER_IFACE = 'org.bluez.AgentManager1'
 AGENT_PATH = '/io/bluetooth_speaker/agent'
-CAPABILITY = 'NoInputNoOutput'
+CAPABILITY = 'DisplayYesNo'
 CALLS_SERVICE_UUID = '0000111e-0000-1000-8000-00805f9b34fb'
 AUDIO_SERVICE_UUID = '0000110d-0000-1000-8000-00805f9b34fb'
 
@@ -94,13 +94,13 @@ async def trust_and_connect_device(device_path):
 
         device_iface = device_obj.get_interface(DEVICE_IFACE)
 
-        try:
-            logging.info(f"Attempting to disconnect {device_path} before connecting.")
-            await device_iface.call_disconnect()
-            await asyncio.sleep(1) # Give it a moment to settle
-        except DBusError as e:
-            # This is expected if the device is not already connected.
-            logging.info(f"Disconnect failed (this is likely okay): {e}")
+        #try:
+        #    logging.info(f"Attempting to disconnect {device_path} before connecting.")
+        #    await device_iface.call_disconnect()
+        #    await asyncio.sleep(1) # Give it a moment to settle
+        #except DBusError as e:
+        #    # This is expected if the device is not already connected.
+        #    logging.info(f"Disconnect failed (this is likely okay): {e}")
 
         await device_iface.call_connect()
         logging.info(f'Connected to device {device_path}')