diff --git a/packages/antlion/controllers/access_point.py b/packages/antlion/controllers/access_point.py
index bdb6923..477f6c8 100755
--- a/packages/antlion/controllers/access_point.py
+++ b/packages/antlion/controllers/access_point.py
@@ -17,7 +17,7 @@
 import ipaddress
 import time
 from dataclasses import dataclass
-from typing import Any, Dict, FrozenSet, List, Literal, Optional, Set, Tuple
+from typing import Any, Dict, FrozenSet, List, Optional, Set, Tuple
 
 from antlion import logger, utils
 from antlion.capabilities.ssh import SSHConfig, SSHProvider
@@ -751,7 +751,7 @@
     force_wmm: Optional[bool] = None,
     hidden: Optional[bool] = False,
     security: Optional[Security] = None,
-    pmf_support: Optional[Literal[1, 2, 3]] = None,
+    pmf_support: Optional[int] = None,
     additional_ap_parameters: Optional[Dict[str, Any]] = None,
     password: Optional[str] = None,
     n_capabilities: Optional[List[Any]] = None,
diff --git a/packages/antlion/controllers/ap_lib/hostapd_ap_preset.py b/packages/antlion/controllers/ap_lib/hostapd_ap_preset.py
index 1875c8b..cf3aab6 100644
--- a/packages/antlion/controllers/ap_lib/hostapd_ap_preset.py
+++ b/packages/antlion/controllers/ap_lib/hostapd_ap_preset.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, FrozenSet, List, Literal, Optional, TypeVar
+from typing import Any, FrozenSet, List, Optional, TypeVar
 
 from antlion import utils
 from antlion.controllers.ap_lib import hostapd_config, hostapd_constants, hostapd_utils
@@ -51,7 +51,7 @@
     mode: Optional[str] = None,
     frequency: Optional[int] = None,
     security: Optional[Security] = None,
-    pmf_support: Optional[Literal[1, 2, 3]] = None,
+    pmf_support: Optional[int] = None,
     ssid: Optional[str] = None,
     hidden: Optional[bool] = None,
     dtim_period: Optional[int] = None,
diff --git a/packages/antlion/controllers/ap_lib/hostapd_config.py b/packages/antlion/controllers/ap_lib/hostapd_config.py
index 51744d8..75c7229 100644
--- a/packages/antlion/controllers/ap_lib/hostapd_config.py
+++ b/packages/antlion/controllers/ap_lib/hostapd_config.py
@@ -14,7 +14,7 @@
 
 import collections
 import logging
-from typing import Any, Dict, FrozenSet, List, Literal, Optional, Union
+from typing import Any, Dict, FrozenSet, List, Optional, Union
 
 from antlion.controllers.ap_lib import hostapd_constants
 from antlion.controllers.ap_lib.hostapd_bss_settings import BssSettings
@@ -96,7 +96,7 @@
         security: Optional[Security] = None,
         bssid: Optional[str] = None,
         force_wmm: Optional[bool] = None,
-        pmf_support: Optional[Literal[1, 2, 3]] = None,
+        pmf_support: Optional[int] = None,
         obss_interval: Optional[int] = None,
         vht_channel_width: Optional[Any] = None,
         vht_center_channel: Optional[int] = None,