diff --git a/visionforge-gdml/src/commonMain/kotlin/space/kscience/visionforge/gdml/markLayers.kt b/visionforge-gdml/src/commonMain/kotlin/space/kscience/visionforge/gdml/markLayers.kt
index a47374d0..dfee104e 100644
--- a/visionforge-gdml/src/commonMain/kotlin/space/kscience/visionforge/gdml/markLayers.kt
+++ b/visionforge-gdml/src/commonMain/kotlin/space/kscience/visionforge/gdml/markLayers.kt
@@ -76,15 +76,14 @@ public fun SolidGroup.markLayers(thresholds: List<Int> = listOf(500, 1000, 20000
             if (layerIndex == 0) break
 
             node.vision.layer = layerIndex
-            val removedCount = node.selfCount * node.children.size
+            remaining -= node.selfCount * node.childrenCount
             logger?.apply {
                 if (node.selfCount > 1) {
-                    info { "Prototype with name ${node.name} moved to layer $layerIndex" }
+                    info { "Prototype with name ${node.name} moved to layer $layerIndex. $remaining nodes remains" }
                 } else {
-                    info { "Vision with name ${node.name} moved to layer $layerIndex" }
+                    info { "Vision with name ${node.name} moved to layer $layerIndex. $remaining nodes remains" }
                 }
             }
-            remaining -= removedCount
         }
     }
 }
\ No newline at end of file