diff --git a/monkey/monkey_island/cc/models/machine.py b/monkey/monkey_island/cc/models/machine.py
index 77000c39a..359786d05 100644
--- a/monkey/monkey_island/cc/models/machine.py
+++ b/monkey/monkey_island/cc/models/machine.py
@@ -1,7 +1,8 @@
 from ipaddress import IPv4Interface
-from typing import Optional, Sequence, TypeAlias
+from typing import Optional, Sequence
 
 from pydantic import Field, PositiveInt, validator
+from typing_extensions import TypeAlias
 
 from common import OperatingSystem
 from common.base_models import MutableInfectionMonkeyBaseModel