diff --git a/test/integration/ITensorNetworks_test.jl b/test/integration/ITensorNetworks_test.jl
index 89beed1a..e86a14ff 100644
--- a/test/integration/ITensorNetworks_test.jl
+++ b/test/integration/ITensorNetworks_test.jl
@@ -1,6 +1,6 @@
 @testset "ITensorNetworks" begin
     using ITensors: ITensors, ITensor, Index, array
-    using ITensorNetworks: ITensorNetwork
+    using ITensorNetworks: ITensorNetwork, vertices
 
     i = Index(2, "i")
     j = Index(3, "j")
@@ -19,7 +19,7 @@
 
     itn = convert(ITensorNetwork, tn)
     @test itn isa ITensorNetwork
-    @test issetequal(arrays(itn), array.([a, b, c]))
+    @test issetequal(map(v -> array(itn[v]), vertices(itn)), array.([a, b, c]))
 
     # TODO test Quantum
 end
diff --git a/test/integration/ITensors_test.jl b/test/integration/ITensors_test.jl
index 0ec30269..b2e80f3b 100644
--- a/test/integration/ITensors_test.jl
+++ b/test/integration/ITensors_test.jl
@@ -17,7 +17,10 @@
     @test itensor isa ITensor
     @test size(itensor) == (2, 3, 4)
     @test array(itensor) == parent(tensor)
-    @test all(splat(==), zip(map(x -> replace(x, "\"" => ""), string.(tags.(ITensors.inds(itensor)))), ["i", "j", "k"]))
+    @test all(
+        splat(==),
+        zip(map(x -> replace(x, "\"" => ""), string.(ITensors.tags.(ITensors.inds(itensor)))), ["i", "j", "k"]),
+    )
 
     tn = rand(TensorNetwork, 4, 3)
     itensors = convert(Vector{ITensor}, tn)