diff --git a/pkg/drivers/cdp/network/manager.go b/pkg/drivers/cdp/network/manager.go index 2caaf222..14d07130 100644 --- a/pkg/drivers/cdp/network/manager.go +++ b/pkg/drivers/cdp/network/manager.go @@ -494,17 +494,16 @@ func (m *Manager) NavigateBack(ctx context.Context, skip values.Int) (values.Boo } func (m *Manager) WaitForNavigation(ctx context.Context, opts WaitEventOptions) error { - ctx, cancel := context.WithCancel(ctx) - defer cancel() - stream, err := m.OnNavigation(ctx) - if err != nil { return err } defer stream.Close(ctx) + ctx, cancel := context.WithCancel(ctx) + defer cancel() + for evt := range stream.Read(ctx) { if err := ctx.Err(); err != nil { return err