diff --git a/pkg/discovery.go b/pkg/discovery.go
index cffcee8..9519ccf 100644
--- a/pkg/discovery.go
+++ b/pkg/discovery.go
@@ -264,7 +264,8 @@ func (d *DiscoveryClient) GetHostStateViaMetadata(ctx context.Context, addr, pri
 	defer conn.Close()
 
 	c := proto.NewBlobCacheClient(conn)
-	resp, err := c.GetState(ctx, &proto.GetStateRequest{})
+
+	resp, err := c.GetState(dialCtx, &proto.GetStateRequest{})
 	if err != nil {
 		return nil, err
 	}