diff --git a/lib/Service/ProvidersAI/TaskProcessingService.php b/lib/Service/ProvidersAI/TaskProcessingService.php
index 64bd1282..d8672af5 100644
--- a/lib/Service/ProvidersAI/TaskProcessingService.php
+++ b/lib/Service/ProvidersAI/TaskProcessingService.php
@@ -384,7 +384,7 @@ public function getInputShape(): array {
 				return array_map(static fn (array $shape) => new ShapeDescriptor(
 					$shape['name'],
 					$shape['description'],
-					EShapeType::from($shape['type']),
+					EShapeType::from($shape['shape_type']),
 				), $this->customTaskType['input_shape']);
 			}
 
@@ -392,7 +392,7 @@ public function getOutputShape(): array {
 				return array_map(static fn (array $shape) => new ShapeDescriptor(
 					$shape['name'],
 					$shape['description'],
-					EShapeType::from($shape['type']),
+					EShapeType::from($shape['shape_type']),
 				), $this->customTaskType['output_shape']);
 			}
 		};