diff --git a/cms/djangoapps/contentstore/management/commands/migrate_cert_config.py b/cms/djangoapps/contentstore/management/commands/migrate_cert_config.py
index 326debf0f7ba..b27dca555e11 100644
--- a/cms/djangoapps/contentstore/management/commands/migrate_cert_config.py
+++ b/cms/djangoapps/contentstore/management/commands/migrate_cert_config.py
@@ -67,7 +67,7 @@ def _parse_course_key(self, raw_value: str) -> CourseKey:
 
         return result
 
-    def get_mongo_courses(self):
+    def get_mongo_courses(self) -> Iterator[CourseKey]:
         """
         Return objects for any mongo(old approach) modulestore backend course that has a certificate configuration.
         """
@@ -87,7 +87,7 @@ def get_mongo_courses(self):
                 course['_id']['name'],
             )
 
-    def get_split_courses(self):
+    def get_split_courses(self) -> Iterator[CourseKey]:
         """
         Return objects for any split modulestore backend course that has a certificate configuration.
         """