diff --git a/lib/has_many_polymorphs/railtie.rb b/lib/has_many_polymorphs/railtie.rb index 6633512..f16f901 100644 --- a/lib/has_many_polymorphs/railtie.rb +++ b/lib/has_many_polymorphs/railtie.rb @@ -3,7 +3,7 @@ class Railtie < Rails::Railtie initializer "has_many_polymorphs" do |app| require 'has_many_polymorphs/autoload' if defined?(Spork) - Spork.trap_class_methods(HasManyPolymorphs, :setup) + Spork.trap_class_method(HasManyPolymorphs, :setup) end HasManyPolymorphs.setup end