diff --git a/monkey/monkey_island/cc/models/machine.py b/monkey/monkey_island/cc/models/machine.py index 1c4c4009c..8e2c0d122 100644 --- a/monkey/monkey_island/cc/models/machine.py +++ b/monkey/monkey_island/cc/models/machine.py @@ -4,6 +4,7 @@ from typing import Optional, Sequence from pydantic import Field, PositiveInt, validator from common import OperatingSystem +from common.types import HardwareID from .base_models import MutableBaseModel from .transforms import make_immutable_sequence @@ -13,7 +14,7 @@ MachineID = PositiveInt class Machine(MutableBaseModel): id: MachineID = Field(..., allow_mutation=False) - hardware_id: Optional[PositiveInt] + hardware_id: Optional[HardwareID] network_interfaces: Sequence[IPv4Interface] operating_system: OperatingSystem operating_system_version: str