diff --git a/pkg/cmd/codespace/common.go b/pkg/cmd/codespace/common.go
index e4569bf43..399c488f8 100644
--- a/pkg/cmd/codespace/common.go
+++ b/pkg/cmd/codespace/common.go
@@ -57,10 +57,7 @@ func (a *App) StopProgressIndicator() {
 }
 
 func (a *App) RunWithProgress(label string, f func() error) error {
-	a.StartProgressIndicatorWithLabel(label)
-	defer a.StopProgressIndicator()
-
-	return f()
+	return a.io.RunWithProgress(label, f)
 }
 
 // Connects to a codespace using Live Share and returns that session
diff --git a/pkg/iostreams/iostreams.go b/pkg/iostreams/iostreams.go
index eeb861938..39e879625 100644
--- a/pkg/iostreams/iostreams.go
+++ b/pkg/iostreams/iostreams.go
@@ -305,6 +305,13 @@ func (s *IOStreams) StopProgressIndicator() {
 	s.progressIndicator = nil
 }
 
+func (s *IOStreams) RunWithProgress(label string, f func() error) error {
+	s.StartProgressIndicatorWithLabel(label)
+	defer s.StopProgressIndicator()
+
+	return f()
+}
+
 func (s *IOStreams) StartAlternateScreenBuffer() {
 	if s.alternateScreenBufferEnabled {
 		s.alternateScreenBufferMu.Lock()