diff --git a/leetcode-ruby.gemspec b/leetcode-ruby.gemspec
index bf51dc15..c2cc5f5b 100644
--- a/leetcode-ruby.gemspec
+++ b/leetcode-ruby.gemspec
@@ -5,7 +5,7 @@ require 'English'
 ::Gem::Specification.new do |s|
   s.required_ruby_version = '>= 3.0'
   s.name = 'leetcode-ruby'
-  s.version = '6.9.4'
+  s.version = '6.9.4.1'
   s.license = 'MIT'
   s.files = ::Dir['lib/**/*.rb'] + %w[README.md]
   s.executable = 'leetcode-ruby'
diff --git a/test/medium/test_380_insert_delete_getrandom_o1.rb b/test/medium/test_380_insert_delete_getrandom_o1.rb
index b646e7a7..7a8e35c3 100644
--- a/test/medium/test_380_insert_delete_getrandom_o1.rb
+++ b/test/medium/test_380_insert_delete_getrandom_o1.rb
@@ -10,7 +10,10 @@ def test_default_one
     assert(randomized_set.insert(1))
     assert(!randomized_set.remove(2))
     assert(randomized_set.insert(2))
-    assert(randomized_set.get_random == 1 || randomized_set.get_random == 2)
+
+    rand = randomized_set.get_random
+    assert([1, 2].include?(rand))
+
     assert(randomized_set.remove(1))
     assert(!randomized_set.insert(2))
     assert_equal(2, randomized_set.get_random)