diff --git a/scripts/task_scheduler.py b/scripts/task_scheduler.py
index 261e5a8..09fb586 100644
--- a/scripts/task_scheduler.py
+++ b/scripts/task_scheduler.py
@@ -21,6 +21,7 @@ from typing import Callable, Optional
 
 from circuitforge_core.tasks.scheduler import (
     LocalScheduler as _CoreTaskScheduler,
+    TaskSpec,  # noqa: F401 — re-exported as part of public API; tests import from here
 )
 
 logger = logging.getLogger(__name__)