diff --git a/packages/grafeas/tests/unit/gapic/grafeas_v1/test_grafeas.py b/packages/grafeas/tests/unit/gapic/grafeas_v1/test_grafeas.py
index 18e4aa9e0d7a..c3a8a52e69ad 100644
--- a/packages/grafeas/tests/unit/gapic/grafeas_v1/test_grafeas.py
+++ b/packages/grafeas/tests/unit/gapic/grafeas_v1/test_grafeas.py
@@ -27,7 +27,7 @@
 import math
 import pytest
 from proto.marshal.rules.dates import DurationRule, TimestampRule
-
+from proto.marshal.rules import wrappers
 
 from google.api_core import client_options
 from google.api_core import exceptions as core_exceptions