diff --git a/frontend/src/lib/inferArgSig.ts b/frontend/src/lib/inferArgSig.ts
index c54cce752a..ec4fcdd819 100644
--- a/frontend/src/lib/inferArgSig.ts
+++ b/frontend/src/lib/inferArgSig.ts
@@ -81,6 +81,9 @@ export function argSigToJsonSchemaType(
 		if (t.str) {
 			newS.originalType = 'enum'
 			newS.enum = t.str
+		} else if (oldS.originalType == 'string' && oldS.enum) {
+			newS.originalType = 'string'
+			newS.enum = oldS.enum
 		} else {
 			newS.originalType = 'string'
 			newS.enum = undefined