diff --git a/public/mod/assign/tests/generator/behat_mod_assign_generator.php b/public/mod/assign/tests/generator/behat_mod_assign_generator.php
index 8e3fd587e70..3f509ca322a 100644
--- a/public/mod/assign/tests/generator/behat_mod_assign_generator.php
+++ b/public/mod/assign/tests/generator/behat_mod_assign_generator.php
@@ -35,7 +35,7 @@ class behat_mod_assign_generator extends behat_generator_base {
                 'singular' => 'submission',
                 'datagenerator' => 'submission',
                 'required' => ['assign', 'user'],
-                'switchids' => ['assign' => 'assignid', 'user' => 'userid'],
+                'switchids' => ['assign' => 'cmid', 'user' => 'userid'],
             ],
             'extensions' => [
                 'singular' => 'extension',