Wrap elastic_client_setup initializer in `ee` block

config/initializers/elastic_client_setup.rb

@@ -2,11 +2,12 @@
 
 require 'gitlab/current_settings'
 
-Elasticsearch::Model::Response::Records.prepend GemExtensions::Elasticsearch::Model::Response::Records
-Elasticsearch::Model::Adapter::Multiple::Records.prepend GemExtensions::Elasticsearch::Model::Adapter::Multiple::Records
-Elasticsearch::Model::Indexing::InstanceMethods.prepend GemExtensions::Elasticsearch::Model::Indexing::InstanceMethods
+Gitlab.ee do
+  Elasticsearch::Model::Response::Records.prepend GemExtensions::Elasticsearch::Model::Response::Records
+  Elasticsearch::Model::Adapter::Multiple::Records.prepend GemExtensions::Elasticsearch::Model::Adapter::Multiple::Records
+  Elasticsearch::Model::Indexing::InstanceMethods.prepend GemExtensions::Elasticsearch::Model::Indexing::InstanceMethods
 
-module Elasticsearch
+  module Elasticsearch
     module Model
       module Client
         # This mutex is only used to synchronize *creation* of a new client, so
@@ -43,4 +44,5 @@ module Elasticsearch
         end
       end
     end
+  end
 end