diff --git a/agents/drivers/xugu/main_test.go b/agents/drivers/xugu/main_test.go index c8085ca585..d302d96a4d 100644 --- a/agents/drivers/xugu/main_test.go +++ b/agents/drivers/xugu/main_test.go @@ -942,8 +942,8 @@ func TestXuguWatchdogFiresKillAndCancel(t *testing.T) { select { case <-killCh: - default: - t.Fatal("killSession was not called when watchdog fired") + case <-time.After(time.Second): + t.Fatal("killSession was not called after watchdog cancellation") } }