should be cancelled here (#390)

Author: tonicmuroq

rpc/rpc.go

@@ -826,7 +826,6 @@ func (v *Vibranium) RunAndWait(stream pb.CoreRPC_RunAndWaitServer) error {
 		// force mark stdin to false
 		opts.OpenStdin = false
 	}
-	defer cancel()
 
 	v.taskAdd("RunAndWait", true)
 
@@ -865,6 +864,7 @@ func (v *Vibranium) RunAndWait(stream pb.CoreRPC_RunAndWaitServer) error {
 	// then deal with the rest messages
 	runAndWait := func(f func(<-chan *types.AttachWorkloadMessage)) {
 		defer v.taskDone("RunAndWait", true)
+		defer cancel()
 		f(ch)
 	}