diff --git a/cluster/node_test.go b/cluster/node_test.go
index 38ad306674..5a0a31af92 100644
--- a/cluster/node_test.go
+++ b/cluster/node_test.go
@@ -49,7 +49,7 @@ func TestNodeSpecs(t *testing.T) {
 
 	assert.Equal(t, node.Cpus, mockInfo.NCPU)
 	assert.Equal(t, node.Memory, mockInfo.MemTotal)
-	assert.Equal(t, node.Labels["graphdriver"], mockInfo.Driver)
+	assert.Equal(t, node.Labels["storagedriver"], mockInfo.Driver)
 	assert.Equal(t, node.Labels["executiondriver"], mockInfo.ExecutionDriver)
 	assert.Equal(t, node.Labels["kernelversion"], mockInfo.KernelVersion)
 	assert.Equal(t, node.Labels["operatingsystem"], mockInfo.OperatingSystem)