diff --git a/ext/civiimport/civiimport.php b/ext/civiimport/civiimport.php
index ded5220c2c45..f1774dd6cfdd 100644
--- a/ext/civiimport/civiimport.php
+++ b/ext/civiimport/civiimport.php
@@ -208,11 +208,13 @@ function civiimport_civicrm_alterTemplateFile($formName, $form, $type, &$templat
  *
  * @param array $tasks
  * @param bool $checkPermissions
- * @param int|null $userId
+ * @param int|null $userID
+ * @param array|null $search
+ * @param array|null $display
  *
  * @noinspection PhpUnused
  */
-function civiimport_civicrm_searchKitTasks(array &$tasks, bool $checkPermissions, ?int $userId) {
+function civiimport_civicrm_searchKitTasks(array &$tasks, bool $checkPermissions, ?int $userID, ?array $search = [], ?array $display = []) {
   foreach (Import::getImportTables() as $import) {
     $tasks['Import_' . $import['user_job_id']]['validate'] = [
       'title' => E::ts('Validate'),