diff --git a/crates/synor-compute/src/scheduler/mod.rs b/crates/synor-compute/src/scheduler/mod.rs
index bd8b50c..225dd5b 100644
--- a/crates/synor-compute/src/scheduler/mod.rs
+++ b/crates/synor-compute/src/scheduler/mod.rs
@@ -515,8 +515,8 @@ pub enum TaskExecutionResult {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::processor::Precision;
-    use crate::task::TaskStatus;
+    use crate::processor::{Operation, Precision};
+    use crate::task::{TaskPriority, TaskStatus};
 
     fn create_test_task(id: u64, op: Operation, deps: Vec<TaskId>) -> Task {
         Task {
diff --git a/crates/synor-compute/src/scheduler/work_queue.rs b/crates/synor-compute/src/scheduler/work_queue.rs
index a961066..fba13d1 100644
--- a/crates/synor-compute/src/scheduler/work_queue.rs
+++ b/crates/synor-compute/src/scheduler/work_queue.rs
@@ -1,7 +1,7 @@
 //! Work queue with thread-safe task management.
 
 use crate::processor::ProcessorType;
-use crate::task::{Task, TaskPriority};
+use crate::task::{Task, TaskId, TaskPriority};
 use crossbeam_channel::{bounded, Receiver, Sender, TryRecvError};
 use std::collections::HashMap;
 use std::sync::atomic::{AtomicU64, Ordering};