more test fixing

Author: AbdulRahmanAlHamali

test/pid_controller_test.rb

@@ -115,19 +115,19 @@ def test_rejection_rate_decreases_with_successful_pings
 
     time_travel(1.0) do
       @controller.update
-    end
-    initial_rejection = @controller.rejection_rate
 
-    # Now simulate successful pings and lower error rate
-    10.times { @controller.record_request(:success) }
-    5.times { @controller.record_ping(:success) }
+      initial_rejection = @controller.rejection_rate
 
-    time_travel(1.0) do
-      @controller.update
-    end
+      # Now simulate successful pings and lower error rate
+      10.times { @controller.record_request(:success) }
+      5.times { @controller.record_ping(:success) }
 
-    # Rejection rate should decrease
-    assert_operator(@controller.rejection_rate, :<, initial_rejection)
+      time_travel(1.0) do
+        @controller.update
+
+        assert_operator(@controller.rejection_rate, :<, initial_rejection)
+      end
+    end
   end
 
   def test_rejection_rate_clamped_between_0_and_1