diff --git a/examples/07_edge_cases.rb b/examples/07_edge_cases.rb
index eb9f338..09b5c2f 100644
--- a/examples/07_edge_cases.rb
+++ b/examples/07_edge_cases.rb
@@ -17,11 +17,11 @@
   chat.user("Hello")
   chat.generate!
   puts "✗ Should have raised an error"
-rescue OpenAI::Errors::BadRequestError => e
-  if e.message.include?("does not exist")
+rescue OpenAI::Errors::NotFoundError => e
+  if e.body.dig(:error, :code) == "model_not_found"
     puts "✓ Invalid model error correctly propagated"
   else
-    puts "✗ Error propagated but message unclear: #{e.message[0..80]}..."
+    puts "✗ Error propagated but code unclear: #{e.body.inspect[0..80]}..."
   end
 rescue => e
   puts "✗ Unexpected error type: #{e.class} - #{e.message[0..80]}..."