diff --git a/badges/tests/behat/manage_badges.feature b/badges/tests/behat/manage_badges.feature
index 6d65c75371e..a058659c9bf 100644
--- a/badges/tests/behat/manage_badges.feature
+++ b/badges/tests/behat/manage_badges.feature
@@ -51,8 +51,8 @@ Feature: Manage badges
 
   Scenario Outline: Filter managed badges
     Given the following "core_badges > Badges" exist:
-      | name     | status | version |
-      | Badge #2 | 1      | 2.0     |
+      | name     | status | version | image                        |
+      | Badge #2 | 1      | 2.0     | badges/tests/behat/badge.png |
     And I log in as "admin"
     When I navigate to "Badges > Manage badges" in site administration
     And I click on "Filters" "button"
@@ -124,9 +124,9 @@ Feature: Manage badges
       | user1    | User      | One      |
       | user2    | User      | Two      |
     And the following "core_badges > Badges" exist:
-      | name     | status |
-      | Badge #2 | 1      |
-      | Badge #3 | 1      |
+      | name     | status | image                        |
+      | Badge #2 | 1      | badges/tests/behat/badge.png |
+      | Badge #3 | 1      | badges/tests/behat/badge.png |
     And the following "core_badges > Issued badges" exist:
       | badge    | user  |
       | Badge #1 | user1 |
diff --git a/badges/tests/generator/behat_core_badges_generator.php b/badges/tests/generator/behat_core_badges_generator.php
index eeb13fb3e56..349c9f4ef6a 100644
--- a/badges/tests/generator/behat_core_badges_generator.php
+++ b/badges/tests/generator/behat_core_badges_generator.php
@@ -37,6 +37,7 @@ class behat_core_badges_generator extends behat_generator_base {
                 'datagenerator' => 'badge',
                 'required' => [
                     'name',
+                    'image',
                 ],
                 'switchids' => [
                     'course' => 'courseid',