diff --git a/test/graph_creation_v2_test.go b/test/graph_creation_v2_test.go
index 3d20dd04..bfd41e94 100644
--- a/test/graph_creation_v2_test.go
+++ b/test/graph_creation_v2_test.go
@@ -128,6 +128,7 @@ func TestGraphCreationV2(t *testing.T) {
 		options, collections := newGraphOpts(db)
 
 		options.ReplicationFactor = driver.SatelliteGraph
+		options.NumberOfShards = 1
 		options.IsSmart = false
 		options.SmartGraphAttribute = ""
 
@@ -149,6 +150,7 @@ func TestGraphCreationV2(t *testing.T) {
 		options, collections := newGraphOpts(db)
 
 		options.ReplicationFactor = driver.SatelliteGraph
+		options.NumberOfShards = 1
 		options.IsSmart = false
 		options.SmartGraphAttribute = ""