Rakefile in rbs-0.3.1 vs Rakefile in rbs-0.4.0

@@ -43,10 +43,10 @@
 
     path = Pathname("test/stdlib/#{klass}_test.rb")
     raise "#{path} already exists!" if path.exist?
 
     require "erb"
-    require "ruby/signature"
+    require "rbs"
 
     class TestTemplateBuilder
       attr_reader :klass, :env
 
       def initialize(klass)