diff --git a/caikit_tgis_backend/managed_tgis_subprocess.py b/caikit_tgis_backend/managed_tgis_subprocess.py
index 10f9c73..b6405c3 100644
--- a/caikit_tgis_backend/managed_tgis_subprocess.py
+++ b/caikit_tgis_backend/managed_tgis_subprocess.py
@@ -29,7 +29,7 @@
 import requests
 
 # First Party
-from caikit.core.toolkit.errors import error_handler
+from caikit.core.exceptions import error_handler
 import alog
 
 # Local
diff --git a/caikit_tgis_backend/tgis_backend.py b/caikit_tgis_backend/tgis_backend.py
index a0a4cc7..7ab5347 100644
--- a/caikit_tgis_backend/tgis_backend.py
+++ b/caikit_tgis_backend/tgis_backend.py
@@ -22,9 +22,9 @@
 import grpc
 
 # First Party
+from caikit.core.exceptions import error_handler
 from caikit.core.module_backends.backend_types import register_backend_type
 from caikit.core.module_backends.base import BackendBase
-from caikit.core.toolkit.errors import error_handler
 import alog
 
 # Local
diff --git a/caikit_tgis_backend/tgis_connection.py b/caikit_tgis_backend/tgis_connection.py
index 7d59dbd..f63bdf7 100644
--- a/caikit_tgis_backend/tgis_connection.py
+++ b/caikit_tgis_backend/tgis_connection.py
@@ -23,7 +23,7 @@
 import grpc
 
 # First Party
-from caikit.core.toolkit.errors import error_handler
+from caikit.core.exceptions import error_handler
 import alog
 
 # Local