diff --git a/lib/testing/generator/block_generator.php b/lib/testing/generator/block_generator.php
index 711a4ae8e55..57a4644fedd 100644
--- a/lib/testing/generator/block_generator.php
+++ b/lib/testing/generator/block_generator.php
@@ -23,6 +23,7 @@
  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
  */
 
+defined('MOODLE_INTERNAL') || die();
 
 /**
  * Block generator base class.
diff --git a/lib/testing/generator/component_generator_base.php b/lib/testing/generator/component_generator_base.php
index 07669148149..bdcdb113402 100644
--- a/lib/testing/generator/component_generator_base.php
+++ b/lib/testing/generator/component_generator_base.php
@@ -23,6 +23,7 @@
  * @license   http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
  */
 
+defined('MOODLE_INTERNAL') || die();
 
 /**
  * Component generator base class.
diff --git a/lib/testing/generator/data_generator.php b/lib/testing/generator/data_generator.php
index a3b61146907..6a436d8fdb1 100644
--- a/lib/testing/generator/data_generator.php
+++ b/lib/testing/generator/data_generator.php
@@ -23,6 +23,7 @@
  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
  */
 
+defined('MOODLE_INTERNAL') || die();
 
 /**
  * Data generator class for unit tests and other tools that need to create fake test sites.
diff --git a/lib/testing/generator/module_generator.php b/lib/testing/generator/module_generator.php
index 44910bd20ac..09b1c2992e3 100644
--- a/lib/testing/generator/module_generator.php
+++ b/lib/testing/generator/module_generator.php
@@ -23,6 +23,7 @@
  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
  */
 
+defined('MOODLE_INTERNAL') || die();
 
 /**
  * Module generator base class.
diff --git a/lib/testing/generator/repository_generator.php b/lib/testing/generator/repository_generator.php
index a02fccda5a6..8e845bbb5d5 100644
--- a/lib/testing/generator/repository_generator.php
+++ b/lib/testing/generator/repository_generator.php
@@ -23,6 +23,8 @@
  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
  */
 
+defined('MOODLE_INTERNAL') || die();
+
 /**
  * Repository data generator class
  *
diff --git a/lib/tests/behat/behat_data_generators.php b/lib/tests/behat/behat_data_generators.php
index 2ec9db04fbc..c494ee9c343 100644
--- a/lib/tests/behat/behat_data_generators.php
+++ b/lib/tests/behat/behat_data_generators.php
@@ -26,7 +26,6 @@
 // NOTE: no MOODLE_INTERNAL test here, this file may be required by behat before including /config.php.
 
 require_once(__DIR__ . '/../../behat/behat_base.php');
-require_once(__DIR__ . '/../../testing/generator/lib.php');
 
 use Behat\Gherkin\Node\TableNode as TableNode;
 use Behat\Behat\Exception\PendingException as PendingException;
@@ -127,6 +126,9 @@ class behat_data_generators extends behat_base {
      */
     public function the_following_exists($elementname, TableNode $data) {
 
+        // Now that we need them require the data generators.
+        require_once(__DIR__ . '/../../testing/generator/lib.php');
+
         if (empty(self::$elements[$elementname])) {
             throw new PendingException($elementname . ' data generator is not implemented');
         }