diff --git a/config/initializers/connection_fix.rb b/config/initializers/connection_fix.rb
index e77253af..d15b0c20 100644
--- a/config/initializers/connection_fix.rb
+++ b/config/initializers/connection_fix.rb
@@ -1,10 +1,10 @@
 module ActiveRecord::ConnectionAdapters
-  class MysqlAdapter
+  class Mysql2Adapter
     alias_method :execute_without_retry, :execute
 
     def execute(*args)
       execute_without_retry(*args)
-    rescue ActiveRecord::StatementInvalid => e
+    rescue Mysql2::Error => e
       if e.message =~ /server has gone away/i
         warn "Server timed out, retrying"
         reconnect!
@@ -14,4 +14,4 @@ module ActiveRecord::ConnectionAdapters
       end
     end
   end
-end
+end
\ No newline at end of file