diff --git a/ext/TensorOperationscuTENSORExt.jl b/ext/TensorOperationscuTENSORExt.jl
index e99abc8..9705ed6 100644
--- a/ext/TensorOperationscuTENSORExt.jl
+++ b/ext/TensorOperationscuTENSORExt.jl
@@ -147,7 +147,7 @@ end
 #-------------------------------------------------------------------------------------------
 # Allocator
 #-------------------------------------------------------------------------------------------
-function CUDAAllocator()
+function TO.CUDAAllocator()
     Mout = CUDA.UnifiedMemory
     Min = CUDA.default_memory
     Mtemp = CUDA.default_memory