diff --git a/lms/static/coffee/src/staff_grading/staff_grading.coffee b/lms/static/coffee/src/staff_grading/staff_grading.coffee
index 7e8f39acac..f4831e35c4 100644
--- a/lms/static/coffee/src/staff_grading/staff_grading.coffee
+++ b/lms/static/coffee/src/staff_grading/staff_grading.coffee
@@ -18,7 +18,7 @@ class StaffGradingBackend
 
   mock: (cmd, data) ->
     # Return a mock response to cmd and data
-    # TODO: needs (optional?) arg for problem location
+    # should take a location as an argument
     if cmd == 'get_next'
       @mock_cnt++
       response =