diff --git a/element/coursefield/classes/element.php b/element/coursefield/classes/element.php
index 00a0de6..a1a50a4 100644
--- a/element/coursefield/classes/element.php
+++ b/element/coursefield/classes/element.php
@@ -39,7 +39,7 @@ class element extends \mod_customcert\element {
     /**
      * This function renders the form elements when adding a customcert element.
      *
-     * @param \mod_customcert\edit_element_form $mform the edit_form instance
+     * @param \MoodleQuickForm $mform the edit form instance
      */
     public function render_form_elements($mform) {
         // Get the user profile fields.
@@ -110,7 +110,7 @@ class element extends \mod_customcert\element {
     /**
      * Sets the data on the form when editing an element.
      *
-     * @param \mod_customcert\edit_element_form $mform the edit_form instance
+     * @param \MoodleQuickForm $mform the edit form instance
      */
     public function definition_after_data($mform) {
         if (!empty($this->get_data())) {
diff --git a/element/daterange/classes/element.php b/element/daterange/classes/element.php
index bb45e85..62f7bc8 100644
--- a/element/daterange/classes/element.php
+++ b/element/daterange/classes/element.php
@@ -102,7 +102,7 @@ class element extends \mod_customcert\element {
     /**
      * This function renders the form elements when adding a customcert element.
      *
-     * @param \mod_customcert\edit_element_form $mform the edit_form instance
+     * @param \MoodleQuickForm $mform the edit form instance
      */
     public function render_form_elements($mform) {
         global $COURSE;
@@ -209,7 +209,7 @@ class element extends \mod_customcert\element {
     /**
      * Sets the data on the form when editing an element.
      *
-     * @param \mod_customcert\edit_element_form $mform the edit_form instance
+     * @param \MoodleQuickForm $mform the edit form instance
      */
     public function definition_after_data($mform) {
         if (!empty($this->get_data()) && !$mform->isSubmitted()) {