From 64cde49d946135bdfacbe52100f772fae6f47eda Mon Sep 17 00:00:00 2001
From: Anastasia Katsman <anastasia.katsman@jetbrains.com>
Date: Sat, 7 Mar 2026 20:42:06 +0900
Subject: [PATCH] AT-4228: testframework [starter] fix setLicense returning
 frontend context instead of RemDevContext in case of rem dev

(cherry picked from commit 2eac4b82c824de3664ac7554006a15f01e7504f4)

GitOrigin-RevId: 87af03428ebe72a85a271fc6517c69d8dfb49eb7
---
 .../src/com/intellij/ide/starter/ide/IDETestContext.kt       | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tools/intellij.tools.ide.starter/src/com/intellij/ide/starter/ide/IDETestContext.kt b/tools/intellij.tools.ide.starter/src/com/intellij/ide/starter/ide/IDETestContext.kt
index 0c431d5bd5f4..703c2572cba1 100644
--- a/tools/intellij.tools.ide.starter/src/com/intellij/ide/starter/ide/IDETestContext.kt
+++ b/tools/intellij.tools.ide.starter/src/com/intellij/ide/starter/ide/IDETestContext.kt
@@ -599,8 +599,9 @@ open class IDETestContext(
       logOutput("License is not provided")
       return this
     }
-    this.onRemDevContext {
-      return frontendIDEContext.setLicense(license)
+    onRemDevContext {
+      frontendIDEContext.setLicense(license)
+      return this
     }
 
     val licenseKeyFileName: String = when (this.ide.productCode) {