diff --git a/src/pool.jl b/src/pool.jl
index 45d7ff586a..80d543f94e 100644
--- a/src/pool.jl
+++ b/src/pool.jl
@@ -479,7 +479,7 @@ end
       error("Trying to free $buf from an unrelated context")
     end
 
-    dev = current_device()
+    dev = device()
     if stream_ordered(dev)
       # mark the pool as active
       pool_mark(dev)