diff --git a/container_registry/container_analysis/samples.py b/container_registry/container_analysis/samples.py
index 9a3edc4ebc0..2eeda1cae81 100644
--- a/container_registry/container_analysis/samples.py
+++ b/container_registry/container_analysis/samples.py
@@ -367,7 +367,7 @@ def find_high_severity_vulnerabilities_for_image(resource_url, project_id):
     vulnerabilities = grafeas_client.list_occurrences(project_name, filter_str)
     filtered_list = []
     for v in vulnerabilities:
-        if v.severity == Severity.HIGH or v.severity == Severity.CRITICAL:
+        if v.effective_severity == Severity.HIGH or v.effective_severity == Severity.CRITICAL:
             filtered_list.append(v)
     return filtered_list
 # [END containeranalysis_filter_vulnerability_occurrences]
diff --git a/container_registry/container_analysis/samples_test.py b/container_registry/container_analysis/samples_test.py
index b8eb6c5fbcb..04768836d00 100644
--- a/container_registry/container_analysis/samples_test.py
+++ b/container_registry/container_analysis/samples_test.py
@@ -278,6 +278,7 @@ def test_find_high_severity_vulnerabilities(self):
             'note_name': client.note_path(PROJECT_ID, note_id),
             'resource_uri': self.image_url,
             'vulnerability': {
+                'effective_severity': Severity.CRITICAL,
                 'package_issue': [
                     {
                         'affected_cpe_uri': 'your-uri-here',