diff --git a/app/src/androidTest/java/org/isoron/uhabits/activities/habits/list/views/CheckmarkPanelViewTest.java b/app/src/androidTest/java/org/isoron/uhabits/activities/habits/list/views/CheckmarkPanelViewTest.java
index 84954771b..30e225a7a 100644
--- a/app/src/androidTest/java/org/isoron/uhabits/activities/habits/list/views/CheckmarkPanelViewTest.java
+++ b/app/src/androidTest/java/org/isoron/uhabits/activities/habits/list/views/CheckmarkPanelViewTest.java
@@ -60,6 +60,7 @@ public class CheckmarkPanelViewTest extends BaseViewTest
         view = new CheckmarkPanelView(targetContext);
         view.setHabit(habit);
         view.setCheckmarkValues(checkmarks);
+        view.setButtonCount(4);
         view.setColor(ColorUtils.getAndroidTestColor(7));
 
         measureView(view, dpToPixels(200), dpToPixels(200));