Address remaining comment of ag/25829379

Bug: 305195729
Test: PunchHoleTest
Flag: N/A
Change-Id: I6906826a87b2c850e9cd9cce6a0d1da322492b77
diff --git a/packages/SystemUI/compose/scene/src/com/android/compose/animation/scene/PunchHole.kt b/packages/SystemUI/compose/scene/src/com/android/compose/animation/scene/PunchHole.kt
index 105f572..b346a70 100644
--- a/packages/SystemUI/compose/scene/src/com/android/compose/animation/scene/PunchHole.kt
+++ b/packages/SystemUI/compose/scene/src/com/android/compose/animation/scene/PunchHole.kt
@@ -107,16 +107,13 @@
     }
 
     override fun ContentDrawScope.draw() {
-        val holeSize = size()
-        if (holeSize == Size.Zero) {
-            drawContent()
-            return
-        }
-
         drawContent()
 
-        val offset = offset()
-        translate(offset.x, offset.y) { drawHole(holeSize) }
+        val holeSize = size()
+        if (holeSize != Size.Zero) {
+            val offset = offset()
+            translate(offset.x, offset.y) { drawHole(holeSize) }
+        }
     }
 
     private fun DrawScope.drawHole(size: Size) {