diff --git a/test/integration/api_access.py b/test/integration/api_access.py
index 6fd4b1b..58dcf66 100644
--- a/test/integration/api_access.py
+++ b/test/integration/api_access.py
@@ -10,7 +10,7 @@
 
 from exasol.saas.client import (
     openapi,
-    DATABASE_LIMITS,
+    Limits,
 )
 from exasol.saas.client.openapi.models.status import Status
 from exasol.saas.client.openapi.api.databases import (