diff --git a/app/src/main/java/com/google/firebase/ml/md/java/camera/CameraSourcePreview.java b/app/src/main/java/com/google/firebase/ml/md/java/camera/CameraSourcePreview.java
index ef2a287..5f0e563 100644
--- a/app/src/main/java/com/google/firebase/ml/md/java/camera/CameraSourcePreview.java
+++ b/app/src/main/java/com/google/firebase/ml/md/java/camera/CameraSourcePreview.java
@@ -106,7 +106,7 @@ protected void onLayout(boolean changed, int left, int top, int right, int botto
     int childHeight = (int) (childWidth / previewSizeRatio);
     if (childHeight <= layoutHeight) {
       for (int i = 0; i < getChildCount(); ++i) {
-        getChildAt(i).layout(0, 0, childWidth, childHeight);
+        getChildAt(i).layout(0, 0, childWidth, getHeight());
       }
     } else {
       // When the child view is too tall to be fitted in its parent: If the child view is static
diff --git a/app/src/main/java/com/google/firebase/ml/md/kotlin/camera/CameraSourcePreview.kt b/app/src/main/java/com/google/firebase/ml/md/kotlin/camera/CameraSourcePreview.kt
index f071728..9f36700 100644
--- a/app/src/main/java/com/google/firebase/ml/md/kotlin/camera/CameraSourcePreview.kt
+++ b/app/src/main/java/com/google/firebase/ml/md/kotlin/camera/CameraSourcePreview.kt
@@ -94,7 +94,7 @@ class CameraSourcePreview(context: Context, attrs: AttributeSet) : FrameLayout(c
         val childHeight = (layoutWidth / previewSizeRatio).toInt()
         if (childHeight <= layoutHeight) {
             for (i in 0 until childCount) {
-                getChildAt(i).layout(0, 0, layoutWidth, childHeight)
+                getChildAt(i).layout(0, 0, layoutWidth, height)
             }
         } else {
             // When the child view is too tall to be fitted in its parent: If the child view is