diff --git a/test/utils.jl b/test/utils.jl
index bf34a2ca8..10541dce3 100644
--- a/test/utils.jl
+++ b/test/utils.jl
@@ -61,12 +61,12 @@
     end
 
     @testset "topk_nodes" begin
-        A = [1.0 5.0 9.0; 2.0 6.0 10.0; 3.0 7.0 11.0; 4.0 8.0 12.0]
+        A = [1.0 5.0 9.0 2.0; 2.0 6.0 10.0 1.0; 3.0 7.0 11.0 2.0; 4.0 8.0 12.0 1.0]
         B = [0.318907 0.189981 0.991791;
              0.547022 0.977349 0.680538;
              0.921823 0.35132 0.494715;
              0.451793 0.00704976 0.0189275]
-        g1 = rand_graph(3, 6, ndata = (x = A,))
+        g1 = rand_graph(4, 6, ndata = (x = A,))
         g2 = rand_graph(3, 6, ndata = B)
         output1 = topk_nodes(g1, :x, 2)
         output2 = topk_nodes(g2, :x, 1, sortby = 2)