diff --git a/pkg/action/upgrade_test.go b/pkg/action/upgrade_test.go
index 0a436534f..80e3070e5 100644
--- a/pkg/action/upgrade_test.go
+++ b/pkg/action/upgrade_test.go
@@ -60,9 +60,9 @@ func TestUpgradeRelease_Success(t *testing.T) {
 
 	ctx, done := context.WithCancel(t.Context())
 	res, err := upAction.RunWithContext(ctx, rel.Name, buildChart(), vals)
-	done()
 	req.NoError(err)
 	is.Equal(res.Info.Status, release.StatusDeployed)
+	done()
 
 	// Detecting previous bug where context termination after successful release
 	// caused release to fail.