1 files changed, 4 insertions, 2 deletions
diff --git a/app/models/environment.rb b/app/models/environment.rb
index baba023c92d..24d4f1d8761 100644
--- a/app/models/environment.rb
+++ b/app/models/environment.rb
@@ -1,6 +1,4 @@
 class Environment < ActiveRecord::Base
-  include PrometheusAdapterLocator
-
   # Used to generate random suffixes for the slug
   LETTERS = 'a'..'z'
   NUMBERS = '0'..'9'
@@ -159,6 +157,10 @@ class Environment < ActiveRecord::Base
     prometheus_adapter.query(:additional_metrics_environment, self) if has_metrics?
   end
 
+  def prometheus_adapter
+    @prometheus_adapter ||= Prometheus::AdapterService.new(project, deployment_platform).prometheus_adapter
+  end
+
   def slug
     super.presence || generate_slug
   end