diff --git a/mod/assign/submission/file/tests/behat/file_type_restriction.feature b/mod/assign/submission/file/tests/behat/file_type_restriction.feature
index 6086b13cf61..cab3f8fe51d 100644
--- a/mod/assign/submission/file/tests/behat/file_type_restriction.feature
+++ b/mod/assign/submission/file/tests/behat/file_type_restriction.feature
@@ -35,13 +35,10 @@ Feature: In an assignment, limit submittable file types
     And I press "Save and display"
     And I navigate to "Edit settings" in current page administration
     And the field "Accepted file types" matches value "image/png,spreadsheet"
-    And I set the field "Accepted file types" to ""
-    And I press "Choose"
-    And I set the field "Image files" to "1"
-    And I press "Save changes"
+    And I set the field "Accepted file types" to "image"
     And I press "Save and display"
     And I navigate to "Edit settings" in current page administration
-    Then the field "Accepted file types" matches value "image"
+    Then I should see "Image files"
 
   @javascript @_file_upload
   Scenario: Uploading permitted file types for an assignment