diff --git a/test/test_cudiag.jl b/test/test_cudiag.jl
index d7832bf..6a54fed 100644
--- a/test/test_cudiag.jl
+++ b/test/test_cudiag.jl
@@ -1,10 +1,10 @@
-using ITensors,
-  ITensors.NDTensors,
-  ITensorGPU,
-  LinearAlgebra, # For tr()
-  Combinatorics, # For permutations()
-  CUDA,
-  Test
+using ITensors
+using ITensors.NDTensors: NDTensors
+using ITensorGPU
+using LinearAlgebra: tr
+using Combinatorics: permutations
+using CUDA
+using Test: @test, @test_broken, @testset
 
 @testset "cuITensor $T Contractions" for T in (Float64, ComplexF64)
   mi, mj, mk, ml, ma = 2, 3, 4, 5, 6, 7
diff --git a/test/test_cuitensor.jl b/test/test_cuitensor.jl
index 1f0aa05..8ced8e5 100644
--- a/test/test_cuitensor.jl
+++ b/test/test_cuitensor.jl
@@ -60,7 +60,7 @@ using ITensors,
   @testset "Test CuVector(cuITensor)" begin
     v = CuVector(ones(SType, dim(a)))
     A = cuITensor(v, a)
-    @allowscalar begin
+    CUDA.@allowscalar begin
       @test v == CuVector(A)
     end
   end