diff --git a/archinstall/lib/models/authentication.py b/archinstall/lib/models/authentication.py
index 6acbcc4cd0..6683181323 100644
--- a/archinstall/lib/models/authentication.py
+++ b/archinstall/lib/models/authentication.py
@@ -40,7 +40,7 @@ def json(self) -> U2FLoginConfigSerialization:
 		}
 
 	@staticmethod
-	def parse_arg(args: dict[str, Any]) -> 'U2FLoginConfiguration | None':
+	def parse_arg(args: U2FLoginConfigSerialization) -> 'U2FLoginConfiguration | None':
 		u2f_login_method = args.get('u2f_login_method')
 
 		if u2f_login_method is None: