diff --git a/webview/src/setup/components/App.test.tsx b/webview/src/setup/components/App.test.tsx
index 44b232dddc..736977890d 100644
--- a/webview/src/setup/components/App.test.tsx
+++ b/webview/src/setup/components/App.test.tsx
@@ -165,7 +165,7 @@ describe('App', () => {
`DVC & DVCLive can be auto-installed as packages with ${defaultInterpreter}`
)
).toBeInTheDocument()
- expect(screen.getByText('Install')).toBeInTheDocument()
+ expect(screen.getByText('Install (pip)')).toBeInTheDocument()
})
it('should let the user find another Python interpreter to install DVC when the Python extension is not installed', () => {
@@ -215,7 +215,7 @@ describe('App', () => {
pythonBinPath: 'python'
})
- const button = screen.getByText('Install')
+ const button = screen.getByText('Install (pip)')
fireEvent.click(button)
expect(mockPostMessage).toHaveBeenCalledWith({
diff --git a/webview/src/setup/components/dvc/CliUnavailable.tsx b/webview/src/setup/components/dvc/CliUnavailable.tsx
index 22c7c02880..6692faeb37 100644
--- a/webview/src/setup/components/dvc/CliUnavailable.tsx
+++ b/webview/src/setup/components/dvc/CliUnavailable.tsx
@@ -18,7 +18,7 @@ export const CliUnavailable: React.FC