diff --git a/cloud-info/cloud_info_catchall/share_discovery.py b/cloud-info/cloud_info_catchall/share_discovery.py
index b33df41..4b1aee3 100644
--- a/cloud-info/cloud_info_catchall/share_discovery.py
+++ b/cloud-info/cloud_info_catchall/share_discovery.py
@@ -6,6 +6,7 @@
 
 import fedcloudclient.endpoint as fedcli
 from cloud_info_provider.auth_refreshers.oidc_refresh import OidcRefreshToken
+import yaml
 
 
 class ShareDiscovery: