diff --git a/videointelligence/google/cloud/videointelligence_v1beta1/__init__.py b/videointelligence/google/cloud/videointelligence_v1beta1/__init__.py
index 60b42da4de3fe..9e732b5800bd4 100644
--- a/videointelligence/google/cloud/videointelligence_v1beta1/__init__.py
+++ b/videointelligence/google/cloud/videointelligence_v1beta1/__init__.py
@@ -17,7 +17,7 @@
 from google.cloud.gapic.videointelligence.v1beta1.video_intelligence_service_client import VideoIntelligenceServiceClient
 from google.cloud.gapic.videointelligence.v1beta1 import enums
 
-from google.cloud.gapic.videointelligence_v1beta1 import types
+from google.cloud.videointelligence_v1beta1 import types
 
 
 
diff --git a/videointelligence/google/cloud/videointelligence_v1beta1/types.py b/videointelligence/google/cloud/videointelligence_v1beta1/types.py
index 482b4adad57d6..9ac3b8a6b2a5b 100644
--- a/videointelligence/google/cloud/videointelligence_v1beta1/types.py
+++ b/videointelligence/google/cloud/videointelligence_v1beta1/types.py
@@ -20,7 +20,7 @@
 
 
 names = []
-for name, message in get_messages(video_intelligence_pb2):
+for name, message in get_messages(video_intelligence_pb2).items():
     setattr(sys.modules[__name__], name, message)
     names.append(name)