From ed00d44ae136257118671a341e2c9439e049ba63 Mon Sep 17 00:00:00 2001
From: Gopi Sakshihally Bhuthaiah <bhuthaiah@google.com>
Date: Thu, 8 Aug 2024 08:45:15 +0000
Subject: [PATCH] DH Key compute check modification for OOB Pairing

---
 bumble/smp.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/bumble/smp.py b/bumble/smp.py
index 35809378..212489b8 100644
--- a/bumble/smp.py
+++ b/bumble/smp.py
@@ -764,11 +764,7 @@ class Session:
         self.peer_io_capability = SMP_NO_INPUT_NO_OUTPUT_IO_CAPABILITY
 
         # OOB
-        self.oob_data_flag = 0
-        if pairing_config.oob is not None:
-            if pairing_config.oob.peer_data is not None:
-                self.oob_data_flag = 1
-
+        self.oob_data_flag = 1 if pairing.oob and pairing_config.oob.peer_data else 0
 
         # Set up addresses
         self_address = connection.self_resolvable_address or connection.self_address