diff --git a/packages/SystemUI/colorextraction/src/com/google/android/colorextraction/drawable/GradientDrawable.java b/packages/SystemUI/colorextraction/src/com/google/android/colorextraction/drawable/GradientDrawable.java
index e75d95a9226f9..38e16f3d58404 100644
--- a/packages/SystemUI/colorextraction/src/com/google/android/colorextraction/drawable/GradientDrawable.java
+++ b/packages/SystemUI/colorextraction/src/com/google/android/colorextraction/drawable/GradientDrawable.java
@@ -157,7 +157,10 @@ public class GradientDrawable extends Drawable {
     }
 
     private void buildPaints() {
-        final Rect bounds = mWindowBounds;
+        Rect bounds = mWindowBounds;
+        if (bounds.width() == 0) {
+            return;
+        }
 
         float w = bounds.width();
         float h = bounds.height();
@@ -177,7 +180,10 @@ public class GradientDrawable extends Drawable {
 
     @Override
     public void draw(@NonNull Canvas canvas) {
-        final Rect bounds = mWindowBounds;
+        Rect bounds = mWindowBounds;
+        if (bounds.width() == 0) {
+            throw new IllegalStateException("You need to call setScreenSize before drawing.");
+        }
 
         // Splat each gradient
         float w = bounds.width();