diff --git a/lib/pry-rescue/minitest.rb b/lib/pry-rescue/minitest.rb
index 902c118..ae62bca 100644
--- a/lib/pry-rescue/minitest.rb
+++ b/lib/pry-rescue/minitest.rb
@@ -4,7 +4,12 @@
 # minitest, we need to add Exception to its passthrough types
 # Note: We need to check the explicit minitest version because the minitest ecosystem
 # may redefine Minitest::Test for Minitest versions < 5.
-if defined?(Minitest::Test) && Minitest::Unit::VERSION.split('.').first.to_i >= 5
+minitest_version = if defined?(Minitest::Unit)
+  MiniTest::Unit::VERSION
+else
+  Minitest::VERSION
+end
+if defined?(Minitest::Test) && Gem::Version.new(minitest_version) >= Gem::Version.new('5.0.0')
 
   class Minitest::Test
     alias_method :run_without_rescue, :run