diff --git a/spec/ruby_llm_spec.rb b/spec/ruby_llm_spec.rb
index 7615d3995..57f6bf160 100644
--- a/spec/ruby_llm_spec.rb
+++ b/spec/ruby_llm_spec.rb
@@ -8,14 +8,15 @@
     let(:log_file) { double }
     let(:log_level) { double }
 
-    before do
-      described_class.instance_variable_set(:@config, nil)
-      described_class.instance_variable_set(:@logger, nil)
-    end
-
-    after do
+    around do |example|
+      original_config = described_class.instance_variable_get(:@config)
+      original_logger = described_class.instance_variable_get(:@logger)
       described_class.instance_variable_set(:@config, nil)
       described_class.instance_variable_set(:@logger, nil)
+      example.run
+    ensure
+      described_class.instance_variable_set(:@config, original_config)
+      described_class.instance_variable_set(:@logger, original_logger)
     end
 
     context 'with configuration options' do