diff --git a/pkg/cmd/codespace/ports.go b/pkg/cmd/codespace/ports.go
index 4109488f9..f671c2449 100644
--- a/pkg/cmd/codespace/ports.go
+++ b/pkg/cmd/codespace/ports.go
@@ -309,11 +309,12 @@ func (a *App) UpdatePortVisibility(ctx context.Context, codespaceName string, ar
 		})
 
 		// wait for success or failure
-		if err := g.Wait(); err != nil {
+		err := g.Wait()
+		a.StopProgressIndicator()
+		if err != nil {
 			return err
 		}
 
-		a.StopProgressIndicator()
 	}
 
 	return nil