From 156d5845d18bf73ad6f9fe19d514b09d702829b5 Mon Sep 17 00:00:00 2001
From: Alex Dadgar <alex.dadgar@gmail.com>
Date: Tue, 15 Dec 2015 18:08:53 -0800
Subject: [PATCH] Fix api test

---
 api/compose_test.go | 1 -
 api/tasks_test.go   | 5 ++---
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/api/compose_test.go b/api/compose_test.go
index bc76895a9..f4b21fcbd 100644
--- a/api/compose_test.go
+++ b/api/compose_test.go
@@ -69,7 +69,6 @@ func TestCompose(t *testing.T) {
 						Operand: "=",
 					},
 				},
-				RestartPolicy: NewRestartPolicy(),
 				Tasks: []*Task{
 					&Task{
 						Name:   "task1",
diff --git a/api/tasks_test.go b/api/tasks_test.go
index 75f29996d..bbdf12550 100644
--- a/api/tasks_test.go
+++ b/api/tasks_test.go
@@ -8,9 +8,8 @@ import (
 func TestTaskGroup_NewTaskGroup(t *testing.T) {
 	grp := NewTaskGroup("grp1", 2)
 	expect := &TaskGroup{
-		Name:          "grp1",
-		Count:         2,
-		RestartPolicy: NewRestartPolicy(),
+		Name:  "grp1",
+		Count: 2,
 	}
 	if !reflect.DeepEqual(grp, expect) {
 		t.Fatalf("expect: %#v, got: %#v", expect, grp)