diff --git a/registration.module b/registration.module
index 75748b1..63a58d9 100644
--- a/registration.module
+++ b/registration.module
@@ -765,7 +765,7 @@ function registration_event_count($entity_type, $entity_id, $registration_id = N
     if ($registration_id != NULL) {
       $query->condition('registration_id', $registration_id, '<>');
     }
-    $active_held_states = registration_states(array('held' => TRUE, 'active' => TRUE));
+    $active_held_states = array_merge(registration_get_active_states(), registration_get_held_states());
     if (!empty($active_held_states)) {
       $query->condition('state', $active_held_states, 'IN');
     }