Fix

Author: siakmun-aws

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/session/CodeGenerationState.kt

@@ -85,8 +85,8 @@ class CodeGenerationState(
             codeGenerationRemainingIterationCount = codeGenerationResult.codeGenerationRemainingIterationCount
             codeGenerationTotalIterationCount = codeGenerationResult.codeGenerationTotalIterationCount
             currentIteration =
-                if (codeGenerationRemainingIterationCount != null || codeGenerationTotalIterationCount != null) {
-                    codeGenerationTotalIterationCount!! - codeGenerationRemainingIterationCount!!
+                if (codeGenerationRemainingIterationCount != null && codeGenerationTotalIterationCount != null) {
+                    codeGenerationTotalIterationCount?.let { total -> codeGenerationRemainingIterationCount?.let { remaining -> total - remaining } }
                 } else {
                     currentIteration?.plus(1)
                 }