diff --git a/cpg-language-python/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonFrontendTest.kt b/cpg-language-python/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonFrontendTest.kt
index 027ec27770..86689a7269 100644
--- a/cpg-language-python/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonFrontendTest.kt
+++ b/cpg-language-python/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonFrontendTest.kt
@@ -795,7 +795,7 @@ class PythonFrontendTest : BaseTest() {
         val whlBody = whlStmt.statement as? Block
         assertNotNull(whlBody)
 
-        val xDeclaration = whlBody.statements[0] as? DeclarationStatement
+        val xDeclaration = whlBody.statements[0] as? AssignExpression
         assertNotNull(xDeclaration)
 
         val ifStatement = whlBody.statements[1] as? IfStatement