From b350fc8f7f720795c3e3a5af34ea02540bb64fd9 Mon Sep 17 00:00:00 2001
From: Diptanu Choudhury <diptanuc@gmail.com>
Date: Mon, 9 Nov 2015 12:44:14 -0800
Subject: [PATCH] Returning once we have an error while saving a task that
 doesn't exist

---
 client/alloc_runner.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/client/alloc_runner.go b/client/alloc_runner.go
index eec492f5a..f74d176ad 100644
--- a/client/alloc_runner.go
+++ b/client/alloc_runner.go
@@ -138,11 +138,11 @@ func (r *AllocRunner) SaveState(taskName string) error {
 	defer r.taskLock.RUnlock()
 	var mErr multierror.Error
 	if taskName != "" {
-		tr, ok := r.tasks[taskName]
-		if !ok {
+		if tr, ok := r.tasks[taskName]; ok {
+			r.saveTaskRunnerState(tr, &mErr)
+		} else {
 			mErr.Errors = append(mErr.Errors, fmt.Errorf("[ERR] client: Task with name %v not found in alloc runner %v", taskName, r.alloc.Name))
 		}
-		r.saveTaskRunnerState(tr, &mErr)
 		return mErr.ErrorOrNil()
 	}
 	for _, tr := range r.tasks {