lib/shellac.rb in shellac-0.0.1 vs lib/shellac.rb in shellac-0.0.2

@@ -1,11 +1,15 @@
 require 'rails'
 require 'klarlack'
 
-class Shellac < Rails::Railtie
+class Railtie < Rails::Railtie
   
-  config.after_initialize do
-    require 'shellac/cache'
-    require 'shellac/actions'
+  config.before_initialize do
+    require File.expand_path('../shellac/cache', __FILE__)
+    require File.expand_path('../shellac/actions', __FILE__)
+  end
+  
+  config.to_prepare do
+    ActionController::Base.send :extend, Shellac::Actions
   end
 
 end
\ No newline at end of file