diff --git a/docs/getting_started/optimizing_tasks.md b/docs/getting_started/optimizing_tasks.md
index 6a2632c74..17cf07ec9 100644
--- a/docs/getting_started/optimizing_tasks.md
+++ b/docs/getting_started/optimizing_tasks.md
@@ -110,9 +110,9 @@ scale our workload:
 
 ```{code-cell} ipython3
 import math
-from typing import Tuple
+from typing import Tuple, List
 
-from flytekit import map_task
+from flytekit import map_task, task, workflow
 
 
 @task