diff --git a/checkers_test.go b/checkers_test.go
index 22dd451..8abc678 100644
--- a/checkers_test.go
+++ b/checkers_test.go
@@ -13,6 +13,7 @@ type MockState struct {
 	EqualsError  bool
 	GetReturn    EntityState
 	GetError     bool
+	AllEntities  []EntityState
 }
 
 func (s MockState) AfterSunrise(_ ...DurationString) bool {
@@ -33,6 +34,9 @@ func (s MockState) Get(eid string) (EntityState, error) {
 	}
 	return s.GetReturn, nil
 }
+func (s MockState) List() ([]EntityState, error) {
+	return s.AllEntities, nil
+}
 func (s MockState) Equals(eid, state string) (bool, error) {
 	if s.EqualsError {
 		return false, errors.New("some error")