diff --git a/usvm-python/usvm-python-main/src/main/kotlin/org/usvm/machine/model/PyModel.kt b/usvm-python/usvm-python-main/src/main/kotlin/org/usvm/machine/model/PyModel.kt
index 0269666143..3ad5057623 100644
--- a/usvm-python/usvm-python-main/src/main/kotlin/org/usvm/machine/model/PyModel.kt
+++ b/usvm-python/usvm-python-main/src/main/kotlin/org/usvm/machine/model/PyModel.kt
@@ -61,7 +61,7 @@ class PyModel(
     @Suppress("UNCHECKED_CAST")
     override fun <Key, Sort : USort> getRegion(regionId: UMemoryRegionId<Key, Sort>): UReadOnlyMemoryRegion<Key, Sort> {
         if (regionId is UArrayRegionId<*, *, *> &&
-            regionId.sort == regionId.sort.uctx.addressSort &&
+            regionId.sort == ctx.addressSort &&
             regionId.arrayType == ArrayType
         ) {
             val region = super.getRegion(