diff --git a/tests/unit/gapic/documentai_v1/test_document_processor_service.py b/tests/unit/gapic/documentai_v1/test_document_processor_service.py
index 82d570f2..2faff032 100644
--- a/tests/unit/gapic/documentai_v1/test_document_processor_service.py
+++ b/tests/unit/gapic/documentai_v1/test_document_processor_service.py
@@ -13,9 +13,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import math
 import os
 
+# try/except added for compatibility with python < 3.8
+try:
+    from unittest import mock
+    from unittest.mock import AsyncMock
+except ImportError:
+    import mock
+
+import math
+
 from google.api_core import (
     future,
     gapic_v1,
@@ -45,7 +53,6 @@
 from google.type import postal_address_pb2  # type: ignore
 import grpc
 from grpc.experimental import aio
-import mock
 from proto.marshal.rules.dates import DurationRule, TimestampRule
 import pytest
 
diff --git a/tests/unit/gapic/documentai_v1beta2/test_document_understanding_service.py b/tests/unit/gapic/documentai_v1beta2/test_document_understanding_service.py
index 3acba2d8..93f3f1fa 100644
--- a/tests/unit/gapic/documentai_v1beta2/test_document_understanding_service.py
+++ b/tests/unit/gapic/documentai_v1beta2/test_document_understanding_service.py
@@ -13,9 +13,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import math
 import os
 
+# try/except added for compatibility with python < 3.8
+try:
+    from unittest import mock
+    from unittest.mock import AsyncMock
+except ImportError:
+    import mock
+
+import math
+
 from google.api_core import (
     future,
     gapic_v1,
@@ -36,7 +44,6 @@
 from google.rpc import status_pb2  # type: ignore
 import grpc
 from grpc.experimental import aio
-import mock
 from proto.marshal.rules.dates import DurationRule, TimestampRule
 import pytest
 
diff --git a/tests/unit/gapic/documentai_v1beta3/test_document_processor_service.py b/tests/unit/gapic/documentai_v1beta3/test_document_processor_service.py
index 458fc7a9..91610b6d 100644
--- a/tests/unit/gapic/documentai_v1beta3/test_document_processor_service.py
+++ b/tests/unit/gapic/documentai_v1beta3/test_document_processor_service.py
@@ -13,9 +13,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import math
 import os
 
+# try/except added for compatibility with python < 3.8
+try:
+    from unittest import mock
+    from unittest.mock import AsyncMock
+except ImportError:
+    import mock
+
+import math
+
 from google.api_core import (
     future,
     gapic_v1,
@@ -45,7 +53,6 @@
 from google.type import postal_address_pb2  # type: ignore
 import grpc
 from grpc.experimental import aio
-import mock
 from proto.marshal.rules.dates import DurationRule, TimestampRule
 import pytest