From 367fee5e2b9452bd38e67698b099e161a6954c19 Mon Sep 17 00:00:00 2001
From: Alex Dadgar <alex.dadgar@gmail.com>
Date: Thu, 21 Jan 2016 12:21:42 -0800
Subject: [PATCH] Fix test

---
 nomad/node_endpoint_test.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/nomad/node_endpoint_test.go b/nomad/node_endpoint_test.go
index e6fc889f8..eef5a8b2f 100644
--- a/nomad/node_endpoint_test.go
+++ b/nomad/node_endpoint_test.go
@@ -357,6 +357,13 @@ func TestClientEndpoint_GetNode(t *testing.T) {
 		t.Fatalf("Bad index: %d %d", resp2.Index, resp.Index)
 	}
 
+	if resp2.Node.ComputedClass == 0 {
+		t.Fatalf("bad ComputedClass: %#v", resp2.Node)
+	}
+
+	// Reset for comparison
+	resp2.Node.ComputedClass = 0
+
 	if !reflect.DeepEqual(node, resp2.Node) {
 		t.Fatalf("bad: %#v %#v", node, resp2.Node)
 	}