diff --git a/integration/build/build_traces_test.go b/integration/build/build_traces_test.go
index 1cbc02f987..8cc0301417 100644
--- a/integration/build/build_traces_test.go
+++ b/integration/build/build_traces_test.go
@@ -56,8 +56,11 @@ func TestBuildkitHistoryTracePropagation(t *testing.T) {
 		<-sub.Context().Done()
 	}()
 
+	d, err := progressui.NewDisplay(&testWriter{t}, progressui.AutoMode, progressui.WithPhase("test"))
+	assert.NilError(t, err)
+
 	eg.Go(func() error {
-		_, err := progressui.DisplaySolveStatus(ctxGo, nil, &testWriter{t}, ch, progressui.WithPhase("test"))
+		_, err := d.UpdateFrom(ctxGo, ch)
 		return err
 	})