diff --git a/stubs/redux/stub/stub_actions b/stubs/redux/stub/stub_actions
index 93ea5db7d..781b33036 100644
--- a/stubs/redux/stub/stub_actions
+++ b/stubs/redux/stub/stub_actions
@@ -218,8 +218,9 @@ class FilterStubsByCustom2 implements PersistUI {
   final String value;
 }
 
-class FilterStubsByClient implements PersistUI {
-  FilterStubsByClient([this.clientId]);
+class FilterStubsByEntity implements PersistUI {
+  FilterStubsByEntity({this.entityId, this.entityType});
 
-  final int clientId;
+  final int entityId;
+  final EntityType entityType;
 }