diff --git a/archinstall/lib/models/authentication.py b/archinstall/lib/models/authentication.py
index d260ddac80..6acbcc4cd0 100644
--- a/archinstall/lib/models/authentication.py
+++ b/archinstall/lib/models/authentication.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from enum import Enum
-from typing import Any, NotRequired, Optional, TypedDict
+from typing import Any, NotRequired, TypedDict
 
 from archinstall.lib.translationhandler import tr
 
@@ -40,7 +40,7 @@ def json(self) -> U2FLoginConfigSerialization:
 		}
 
 	@staticmethod
-	def parse_arg(args: dict[str, Any]) -> Optional['U2FLoginConfiguration']:
+	def parse_arg(args: dict[str, Any]) -> 'U2FLoginConfiguration | None':
 		u2f_login_method = args.get('u2f_login_method')
 
 		if u2f_login_method is None: