diff --git a/Project.toml b/Project.toml
index 832e323b2..4d9b2032c 100644
--- a/Project.toml
+++ b/Project.toml
@@ -21,16 +21,16 @@ SIMD = "fdea26ae-647d-5447-a871-4b548cad5224"
 SumTypes = "8e1ec7a9-0e02-4297-b0fe-6433085c89f2"
 
 [weakdeps]
+CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 Makie = "ee78f7c6-11fb-53f2-987a-cfe4a2b5a57a"
 Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"
 Quantikz = "b0d11df0-eea3-4d79-b4a5-421488cbf74b"
-CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 
 [extensions]
+QuantumCliffordGPUExt = "CUDA"
 QuantumCliffordMakieExt = "Makie"
 QuantumCliffordPlotsExt = "Plots"
 QuantumCliffordQuantikzExt = "Quantikz"
-QuantumCliffordGPUExt = "CUDA"
 
 [compat]
 Combinatorics = "1.0"
diff --git a/test/test_gpu.jl b/test/test_gpu.jl
index 22f16824a..018a5aa34 100644
--- a/test/test_gpu.jl
+++ b/test/test_gpu.jl
@@ -1,7 +1,7 @@
 using QuantumClifford
 using CUDA
 
-@testset "CUDA" begin
+@testset "GPU" begin
     @test begin
         p = P"_IZXY"
         p_gpu = to_gpu(p)
@@ -17,7 +17,7 @@ using CUDA
     @test begin
         s = to_gpu(S"XX XZ")
         op = SingleQubitOperator(sHadamard(1))
-        QuantumCliffordGPU._apply!(s, op) 
+        QuantumClifford._apply!(s, op) # todo rewrite this with normal apply not _apply
         correct = S"ZX ZZ"
         to_cpu(s) == correct
     end
@@ -26,7 +26,7 @@ using CUDA
         s = random_stabilizer(10, 10)
         s_gpu = to_gpu(s)
         op = SingleQubitOperator(sHadamard(5))
-        QuantumCliffordGPU._apply!(s_gpu, op) 
+        QuantumClifford._apply!(s_gpu, op) # todo rewrite this with normal apply not _apply
         QuantumClifford._apply!(s, op)
         to_cpu(s_gpu) == s
     end
@@ -35,7 +35,7 @@ using CUDA
         s = random_stabilizer(10, 10)
         s_gpu = to_gpu(s)
         op = sCNOT(2, 3)
-        QuantumCliffordGPU._apply!(s_gpu, op) 
+        QuantumClifford._apply!(s_gpu, op) # todo rewrite this with normal apply not _apply 
         QuantumClifford._apply!(s, op)
         to_cpu(s_gpu) == s
     end