diff --git a/scripts/build_spec.py b/scripts/build_spec.py
index 114832bc2..16dd0d21d 100644
--- a/scripts/build_spec.py
+++ b/scripts/build_spec.py
@@ -31,7 +31,7 @@ from eth2spec.utils.hash_function import hash
 SSZObject = TypeVar('SSZObject', bound=SSZType)
 '''
 PHASE1_IMPORTS = '''from typing import (
-    Any, Dict, Set, Sequence, MutableSequence, NewType, Tuple, Union, TypeVar
+    Any, Dict, Set, Sequence, MutableSequence, NewType, Optional, Tuple, Union, TypeVar
 )
 from math import (
     log2,
diff --git a/specs/phase0/beacon-chain.md b/specs/phase0/beacon-chain.md
index 478a2daa1..7f2e873bf 100644
--- a/specs/phase0/beacon-chain.md
+++ b/specs/phase0/beacon-chain.md
@@ -788,7 +788,7 @@ def compute_activation_exit_epoch(epoch: Epoch) -> Epoch:
 #### `compute_domain`
 
 ```python
-def compute_domain(domain_type: DomainType, fork_version: Version=None) -> Domain:
+def compute_domain(domain_type: DomainType, fork_version: Optional[Version]=None) -> Domain:
     """
     Return the domain for the ``domain_type`` and ``fork_version``.
     """