diff --git a/grid-client/state/state_test.go b/grid-client/state/state_test.go
index e33b779e..8147928b 100644
--- a/grid-client/state/state_test.go
+++ b/grid-client/state/state_test.go
@@ -236,16 +236,18 @@ func TestLoadK8sFromGrid(t *testing.T) {
 	})
 
 	master := workloads.K8sNode{
-		Name:        "test",
-		Node:        1,
-		DiskSize:    0,
-		PublicIP:    false,
-		Planetary:   true,
-		CPU:         1,
-		Memory:      8,
-		PlanetaryIP: "203:8b0b:5f3e:b859:c36:efdf:ab6e:50cc",
-		IP:          "1.1.1.1",
-		NetworkName: "test",
+		Name:          "test",
+		Node:          1,
+		DiskSize:      0,
+		Flist:         flist,
+		FlistChecksum: flistCheckSum,
+		PublicIP:      false,
+		Planetary:     true,
+		CPU:           1,
+		Memory:        8,
+		PlanetaryIP:   "203:8b0b:5f3e:b859:c36:efdf:ab6e:50cc",
+		IP:            "1.1.1.1",
+		NetworkName:   "test",
 	}
 
 	var Workers []workloads.K8sNode
diff --git a/grid-client/workloads/k8s_test.go b/grid-client/workloads/k8s_test.go
index f3267c7e..e879d89f 100644
--- a/grid-client/workloads/k8s_test.go
+++ b/grid-client/workloads/k8s_test.go
@@ -13,18 +13,20 @@ var flist = "https://hub.grid.tf/tf-official-apps/threefoldtech-k3s-latest.flist
 
 // K8sWorkload to be used in tests
 var K8sWorkload = K8sNode{
-	Name:        "test",
-	Node:        0,
-	DiskSize:    5,
-	PublicIP:    false,
-	PublicIP6:   false,
-	Planetary:   false,
-	ComputedIP:  "",
-	ComputedIP6: "",
-	PlanetaryIP: "",
-	IP:          "",
-	CPU:         2,
-	Memory:      1024,
+	Name:          "test",
+	Node:          0,
+	DiskSize:      5,
+	PublicIP:      false,
+	PublicIP6:     false,
+    Planetary:     false,
+	Flist:         flist,
+	FlistChecksum: "e71ee7421f45392fbbb92309182e3006",
+	ComputedIP:    "",
+	ComputedIP6:   "",
+	PlanetaryIP:   "",
+	IP:            "",
+	CPU:           2,
+	Memory:        1024,
 }
 
 func TestK8sNodeData(t *testing.T) {