diff --git a/packages/variables/parseVariables.ts b/packages/variables/parseVariables.ts
index f4ec2b786..28b573f4e 100644
--- a/packages/variables/parseVariables.ts
+++ b/packages/variables/parseVariables.ts
@@ -73,8 +73,9 @@ const evaluateInlineCode = (
   { variables }: { variables: Variable[] }
 ) => {
   try {
+    const body = parseVariables(variables, { fieldToParse: 'id' })(code)
     return createCodeRunner({ variables })(
-      parseVariables(variables, { fieldToParse: 'id' })(code)
+      body.includes('return ') ? body : `return ${body}`
     )
   } catch (err) {
     return parseVariables(variables)(code)