diff --git a/src/Plugin/RulesAction/BatEventValidation.php b/src/Plugin/RulesAction/BatEventValidation.php
index 120bcbe..80d0a29 100644
--- a/src/Plugin/RulesAction/BatEventValidation.php
+++ b/src/Plugin/RulesAction/BatEventValidation.php
@@ -32,10 +32,10 @@ class BatEventValidation extends RulesActionBase {
   /**
    * Validates a Bat Event.
    *
-   * @param \Drupal\Core\Entity\EntityInterface $event
+   * @param \Drupal\bat_event\Entity\Event $event
    *   The entity to be validated.
    */
-  protected function doExecute(EntityInterface $event) {
+  protected function doExecute(\Drupal\bat_event\Entity\Event $event) {
     $user = \Drupal\user\Entity\User::load(\Drupal::currentUser()->id());
     $event_type = bat_event_type_load($event->bundle());
     // we only want to limit hourly reservations