diff --git a/agenta-web/src/components/Playground/NewVariantModal.tsx b/agenta-web/src/components/Playground/NewVariantModal.tsx
index ed8578c85b..34b5c0542c 100644
--- a/agenta-web/src/components/Playground/NewVariantModal.tsx
+++ b/agenta-web/src/components/Playground/NewVariantModal.tsx
@@ -35,7 +35,7 @@ const NewVariantModal: React.FC<Props> = ({
     const [isInputValid, setIsInputValid] = useState(false)
 
     const handleTemplateVariantChange = (value: string) => {
-        let newValue = value.includes(".") ? value.output.split(".")[0] : value
+        let newValue = value.includes(".") ? value.split(".")[0] : value
         setTemplateVariantName(value)
         setVariantPlaceHolder(`${newValue}`)
         setIsInputValid(newVariantName.trim().length > 0 && value !== "Source Variant")