diff --git a/dhcp-service/spec/lease_spec.rb b/dhcp-service/spec/lease_spec.rb
index ca01717..4572242 100644
--- a/dhcp-service/spec/lease_spec.rb
+++ b/dhcp-service/spec/lease_spec.rb
@@ -14,8 +14,9 @@ def wait_for_leases(expected_count, retries = 5, delay = 2)
     #raise "Expected #{expected_count} leases, got #{db_client[:lease4].count}"
   end
 
-  def wait_for_packet_capture(delay = 5)
-    sleep delay # Wait for the specified delay to ensure packet capture completes
+  def wait_for_packet_capture(process, packets, delay = 5)
+    sleep delay 
+    sleep(packets * 0.1) 
   end
 
   before do