diff --git a/src/raytrace/raytrace_cpu.jl b/src/raytrace/raytrace_cpu.jl
index 1d990ba..d9b1bc0 100644
--- a/src/raytrace/raytrace_cpu.jl
+++ b/src/raytrace/raytrace_cpu.jl
@@ -7,7 +7,7 @@ function traverse_rays_nodes!(bvh, points, directions, src, dst, num_src, ::Noth
 
     # Split computation into contiguous ranges of minimum 100 elements each; if only single thread
     # is needed, inline call
-    tp = TaskPartitioner(num_src, options.num_threads, options.min_traversals_per_thread)
+    tp = AK.TaskPartitioner(num_src, options.num_threads, options.min_traversals_per_thread)
     if tp.num_tasks == 1
         num_dst = traverse_rays_nodes_range!(
             bvh, points, directions,
@@ -99,7 +99,7 @@ function traverse_rays_leaves!(bvh, points, directions, src, intersections, num_
 
     # Split computation into contiguous ranges of minimum 100 elements each; if only single thread
     # is needed, inline call
-    tp = TaskPartitioner(num_src, options.num_threads, options.min_traversals_per_thread)
+    tp = AK.TaskPartitioner(num_src, options.num_threads, options.min_traversals_per_thread)
     if tp.num_tasks == 1
         num_intersections = traverse_rays_leaves_range!(
             bvh, points, directions,
@@ -174,4 +174,4 @@ function traverse_rays_leaves_range!(
         num_written[] = num_dst
         return nothing
     end
-end
\ No newline at end of file
+end