diff --git a/docker-compose.yml b/docker-compose.yml
index e7582f55..7395df7d 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -38,10 +38,10 @@ services:
       dockerfile: infra/dockerfile.train_component
     image: mozilla/bugbug-train-component
 
-  bugbug-train-defect:
+  bugbug-train-defectenhancementtask:
     build:
       context: .
-      dockerfile: infra/dockerfile.train_defect
+      dockerfile: infra/dockerfile.train_defectenhancementtask
     image: mozilla/bugbug-train-defect
 
   bugbug-train-regression:
diff --git a/infra/data-pipeline.yml b/infra/data-pipeline.yml
index 0ae46b53..0e5fa5e3 100644
--- a/infra/data-pipeline.yml
+++ b/infra/data-pipeline.yml
@@ -199,7 +199,7 @@ tasks:
       - rollback-test-task
       - train-tracking
       - train-regression
-      - train-defect
+      - train-defectenhancementtask
       - train-component
     payload:
       capabilities:
diff --git a/infra/dockerfile.train_defect b/infra/dockerfile.train_defectenhancementtask
similarity index 100%
rename from infra/dockerfile.train_defect
rename to infra/dockerfile.train_defectenhancementtask