Update plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/service/CodeWhispererService.kt

Co-authored-by: Richard Li <[email protected]>

Author: andrewyuq

plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/service/CodeWhispererService.kt

@@ -209,9 +209,9 @@ class CodeWhispererService(private val cs: CoroutineScope) : Disposable {
 
     internal suspend fun invokeCodeWhispererInBackground(requestContext: RequestContext): Job {
         val popup = withContext(EDT) {
-            val it = CodeWhispererPopupManager.getInstance().initPopup()
-            Disposer.register(it) { CodeWhispererInvocationStatus.getInstance().finishInvocation() }
-            it
+            CodeWhispererPopupManager.getInstance().initPopup().also {
+                Disposer.register(it) { CodeWhispererInvocationStatus.getInstance().finishInvocation() }
+            }
         }
 
         val workerContexts = mutableListOf<WorkerContext>()