diff --git a/test/Tensor_test.jl b/test/Tensor_test.jl
index b159b371..b38d31e3 100644
--- a/test/Tensor_test.jl
+++ b/test/Tensor_test.jl
@@ -189,8 +189,8 @@
         data = rand(2, 2, 2)
         tensor = Tensor(data, (:i, :j, :k))
 
-        @test Base.IteratorSize(tensor) == Base.HasShape()
-        @test Base.IteratorEltype(tensor) == eltype(data)
+        @test Base.IteratorSize(tensor) == Base.HasShape{3}()
+        @test Base.IteratorEltype(tensor) == Base.HasEltype()
         @test all(x -> ==(x...), zip(tensor, data))
     end