diff --git a/edx_exams/apps/core/management/commands/test/test_bulk_add_course_staff.py b/edx_exams/apps/core/management/commands/test/test_bulk_add_course_staff.py
index 836de3e3..e9cb910c 100644
--- a/edx_exams/apps/core/management/commands/test/test_bulk_add_course_staff.py
+++ b/edx_exams/apps/core/management/commands/test/test_bulk_add_course_staff.py
@@ -51,7 +51,7 @@ def test_add_course_staff_with_existing_user(self):
             assert CourseStaffRole.objects.filter(
                 user=self.user.id,
                 course_id=self.course_id,
-                course_role=self.course_role,
+                role=self.course_role,
             ).exists()
 
     def test_add_course_staff_with_new_user(self):
@@ -66,7 +66,7 @@ def test_add_course_staff_with_new_user(self):
             assert CourseStaffRole.objects.filter(
                 user=user.id,
                 course_id=self.course_id,
-                course_role=self.course_role,
+                role=self.course_role,
             ).exists()
 
     def test_add_course_staff_with_not_default_batch_size(self):