From ff52fbefe46424b5313f4de6410deb6da36dce50 Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Tue, 30 Aug 2022 17:17:19 +0530
Subject: [PATCH] BB: Fix AgentConfiguration logic in MonkeyIslandClient

---
 envs/monkey_zoo/blackbox/island_client/monkey_island_client.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py b/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py
index 03ab56d22..43394f24c 100644
--- a/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py
+++ b/envs/monkey_zoo/blackbox/island_client/monkey_island_client.py
@@ -5,7 +5,6 @@ from typing import Sequence, Union
 
 from bson import json_util
 
-from common.agent_configuration import AgentConfiguration
 from common.credentials import Credentials
 from envs.monkey_zoo.blackbox.island_client.monkey_island_requests import MonkeyIslandRequests
 from envs.monkey_zoo.blackbox.test_configurations.test_configuration import TestConfiguration
@@ -51,7 +50,7 @@ class MonkeyIslandClient(object):
     def _import_config(self, test_configuration: TestConfiguration):
         response = self.requests.put_json(
             "api/agent-configuration",
-            json=AgentConfiguration.to_mapping(test_configuration.agent_configuration),
+            json=test_configuration.agent_configuration.dict(),
         )
         if response.ok:
             LOGGER.info("Configuration is imported.")