diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py
index 00a6cae3..440ff6c4 100644
--- a/archinstall/lib/disk.py
+++ b/archinstall/lib/disk.py
@@ -2,6 +2,7 @@ import glob
 import pathlib
 import re
 from collections import OrderedDict
+from typing import Optional
 
 from .general import *
 from .hardware import hasUEFI