diff --git a/ext/search_kit/CRM/Search/Upgrader.php b/ext/search_kit/CRM/Search/Upgrader.php
index aa37914a90dc..24f573e1c567 100644
--- a/ext/search_kit/CRM/Search/Upgrader.php
+++ b/ext/search_kit/CRM/Search/Upgrader.php
@@ -121,8 +121,7 @@ public function upgrade_1003(): bool {
    */
   public function upgrade_1005(): bool {
     $this->ctx->log->info('Applying update 1005 - add acl_bypass column.');
-    $this->addTask('Add Cancel Button Setting to the Profile', 'addColumn',
-      'civicrm_search_display', 'acl_bypass', "tinyint DEFAULT 0 COMMENT 'Skip permission checks and ACLs when running this display.'");
+    $this->addColumn('civicrm_search_display', 'acl_bypass', "tinyint DEFAULT 0 COMMENT 'Skip permission checks and ACLs when running this display.'");
     return TRUE;
   }
 
@@ -158,7 +157,7 @@ public function upgrade_1006(): bool {
    */
   public function upgrade_1007(): bool {
     $this->ctx->log->info('Applying update 1007 - add SearchSegment table.');
-    if (!CRM_Core_DAO::singleValueQuery("SHOW TABLES LIKE 'civicrm_search_segment'")) {
+    if (!CRM_Core_DAO::checkTableExists('civicrm_search_segment')) {
       $createTable = "
 CREATE TABLE `civicrm_search_segment` (
   `id` int unsigned NOT NULL AUTO_INCREMENT COMMENT 'Unique SearchSegment ID',