1 files changed, 3 insertions, 4 deletions
diff --git a/test/psych/test_yamlstore.rb b/test/psych/test_yamlstore.rb
index c31b33f5ed..5d6fcb79b2 100644
--- a/test/psych/test_yamlstore.rb
+++ b/test/psych/test_yamlstore.rb
@@ -1,12 +1,11 @@
-require 'test/unit'
-require 'psych'
+require 'psych/helper'
 require 'yaml/store'
 require 'tmpdir'
 
 module Psych
   Psych::Store = YAML::Store unless defined?(Psych::Store)
 
-  class YAMLStoreTest < Test::Unit::TestCase
+  class YAMLStoreTest < TestCase
     def setup
       @engine, YAML::ENGINE.yamler = YAML::ENGINE.yamler, 'psych'
       @dir = Dir.mktmpdir("rubytest-file")
@@ -78,7 +77,7 @@ module Psych
     end
 
     def test_writing_inside_readonly_transaction_raises_error
-      assert_raise(PStore::Error) do
+      assert_raises(PStore::Error) do
         @yamlstore.transaction(true) do
           @yamlstore[:foo] = "bar"
         end