diff --git a/tests/datasets/test_entity_factory.py b/tests/datasets/test_entity_factory.py
index d7e0f39d41..79a8cd5cff 100644
--- a/tests/datasets/test_entity_factory.py
+++ b/tests/datasets/test_entity_factory.py
@@ -15,6 +15,7 @@
     EntityKey.DISCOVER,
     EntityKey.EVENTS,
     EntityKey.EAP_SPANS,
+    EntityKey.EAP_SPANS_RPC,
     EntityKey.SPANS_NUM_ATTRS,
     EntityKey.SPANS_STR_ATTRS,
     EntityKey.GROUPASSIGNEE,