diff --git a/src/neuroagent/tools/bluenaas_tool.py b/src/neuroagent/tools/bluenaas_tool.py
index 8908778..4fb7bf3 100644
--- a/src/neuroagent/tools/bluenaas_tool.py
+++ b/src/neuroagent/tools/bluenaas_tool.py
@@ -107,11 +107,6 @@ async def _arun(
     ) -> BaseToolOutput:
         """Run the BlueNaaS tool."""
         logger.info("Running BlueNaaS tool")
-        if not current_injection__stimulus__amplitudes:
-            current_injection__stimulus__amplitudes = [0.1]
-        if not record_from:
-            record_from = [RecordingLocation()]
-
         json_api = self.create_json_api(
             current_injection__inject_to=current_injection__inject_to,
             current_injection__stimulus__stimulus_type=current_injection__stimulus__stimulus_type,
@@ -156,8 +151,12 @@ def create_json_api(
         conditions__max_time: int = 100,
         conditions__time_step: float = 0.05,
         conditions__seed: int = 100,
-    ) -> dict[str, str | int | float]:
+    ) -> dict[str, Any]:
         """Based on the simulation config, create a valid JSON for the API."""
+        if not current_injection__stimulus__amplitudes:
+            current_injection__stimulus__amplitudes = [0.1]
+        if not record_from:
+            record_from = [RecordingLocation()]
         json_api = {
             "currentInjection": {
                 "injectTo": current_injection__inject_to,