diff --git a/openflexure_microscope/common/tasks/__init__.py b/openflexure_microscope/common/tasks/__init__.py
index ce5851df..6cf6dc43 100644
--- a/openflexure_microscope/common/tasks/__init__.py
+++ b/openflexure_microscope/common/tasks/__init__.py
@@ -1,2 +1,14 @@
+__all__ = [
+    "taskify",
+    "tasks",
+    "states",
+    "current_task",
+    "update_task_progress",
+    "cleanup_tasks",
+    "remove_task",
+    "update_task_data",
+    "ThreadTerminationError",
+]
+
 from .pool import tasks, states, current_task, update_task_progress, cleanup_tasks, remove_task, update_task_data, taskify
 from .thread import ThreadTerminationError