diff --git a/amy/emails/signals.py b/amy/emails/signals.py
index bdfa2469f..9d4421412 100644
--- a/amy/emails/signals.py
+++ b/amy/emails/signals.py
@@ -10,6 +10,7 @@
     InstructorDeclinedContext,
     InstructorSignupContext,
     InstructorTrainingApproachingContext,
+    InstructorTrainingCompletedNotBadgedContext,
     PersonsMergedContext,
 )
 
@@ -94,19 +95,19 @@ def __init__(self, *args, **kwargs):
 )
 instructor_training_completed_not_badged_signal = Signal(
     signal_name=INSTRUCTOR_TRAINING_COMPLETED_NOT_BADGED_SIGNAL_NAME,
-    context_type=InstructorTrainingApproachingContext,
+    context_type=InstructorTrainingCompletedNotBadgedContext,
 )
 # Emitted when conditions for the previous signal may have changed and
 # the email should be re-calculated.
 instructor_training_completed_not_badged_update_signal = Signal(
     signal_name=INSTRUCTOR_TRAINING_COMPLETED_NOT_BADGED_SIGNAL_NAME,
-    context_type=InstructorTrainingApproachingContext,
+    context_type=InstructorTrainingCompletedNotBadgedContext,
 )
 # Emitted when conditions for the previous signal may have changed and
 # the email should be cancelled.
 instructor_training_completed_not_badged_remove_signal = Signal(
     signal_name=INSTRUCTOR_TRAINING_COMPLETED_NOT_BADGED_SIGNAL_NAME,
-    context_type=InstructorTrainingApproachingContext,
+    context_type=InstructorTrainingCompletedNotBadgedContext,
 )
 
 ALL_SIGNALS = [item for item in locals().values() if isinstance(item, Signal)]