diff --git a/patchfile.gemspec b/patchfile.gemspec
index 045fe86..0a4305a 100644
--- a/patchfile.gemspec
+++ b/patchfile.gemspec
@@ -104,13 +104,15 @@ end
 #    puts "Permission denied in #{ENV['PWD']}/app/helpers/application_helper.rb"
 #end
 begin
+  require 'mongoid'
   require "#{dir_pwd}/app/models/site"
   site_first = Site.first
   if site_first[:display_title_in_frontend].nil?
     puts 'close display double title in frontend'
     site_first.update_attributes(:display_title_in_frontend=>false) 
   end
-rescue ScriptError => e
+rescue Exception => e
+  puts e
   puts 'unable to close title display in frontend.'
 end
 # Describe your gem and declare its dependencies: