diff --git a/message/templates/add_contact_button.mustache b/message/templates/add_contact_button.mustache
index f94428999d6..c06d6614817 100644
--- a/message/templates/add_contact_button.mustache
+++ b/message/templates/add_contact_button.mustache
@@ -15,7 +15,7 @@
     along with Moodle.  If not, see <http://www.gnu.org/licenses/>.
 }}
 {{!
-    @template core/add_contact_button
+    @template core_message/add_contact_button
 
     Template for the contents of the add contact button on the user's profile page.
 
diff --git a/message/templates/remove_contact_button.mustache b/message/templates/remove_contact_button.mustache
index 6d8fec25e1f..2f542f55c2a 100644
--- a/message/templates/remove_contact_button.mustache
+++ b/message/templates/remove_contact_button.mustache
@@ -15,12 +15,16 @@
     along with Moodle.  If not, see <http://www.gnu.org/licenses/>.
 }}
 {{!
-    @template core/add_contact_button
+    @template core_message/add_contact_button
 
     Template for the contents of the add contact button on the user's profile page.
 
     Context variables required for this template:
-    *
+    * none
+
+    Example context (json):
+    {
+    }
 }}
 <span>
     {{#pix}} t/removecontact, core {{/pix}}