From 999c39beb6c2fbb2be500d4f4212d2e1aa40544c Mon Sep 17 00:00:00 2001
From: jmlefevrevulog <42238482+jmlefevrevulog@users.noreply.github.com>
Date: Mon, 25 Jan 2021 11:58:30 +0100
Subject: [PATCH] Update base.py

---
 mysql_server_has_gone_away/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mysql_server_has_gone_away/base.py b/mysql_server_has_gone_away/base.py
index f14fc19..4d49197 100644
--- a/mysql_server_has_gone_away/base.py
+++ b/mysql_server_has_gone_away/base.py
@@ -26,7 +26,7 @@ def wrapper(self, query, args=None):
                 # Map some error codes to IntegrityError, since they seem to be
                 # misclassified and Django would prefer the more logical place.
                 if e.args[0] in self.codes_for_integrityerror:
-                    raise base.utils.IntegrityError(*tuple(e.args))
+                    raise base.IntegrityError(*tuple(e.args))
                 raise
         return wrapper