diff --git a/system/src/Grav/Common/Flex/Types/UserGroups/UserGroupObject.php b/system/src/Grav/Common/Flex/Types/UserGroups/UserGroupObject.php
index c6aa161ba..046856290 100644
--- a/system/src/Grav/Common/Flex/Types/UserGroups/UserGroupObject.php
+++ b/system/src/Grav/Common/Flex/Types/UserGroups/UserGroupObject.php
@@ -16,6 +16,7 @@ use Grav\Common\Flex\Traits\FlexObjectTrait;
 use Grav\Common\User\Access;
 use Grav\Common\User\Interfaces\UserGroupInterface;
 use Grav\Framework\Flex\FlexObject;
+use Grav\Framework\Flex\Traits\FlexMediaTrait;
 
 /**
  * Flex User Group
@@ -29,6 +30,7 @@ class UserGroupObject extends FlexObject implements UserGroupInterface
 {
     use FlexGravTrait;
     use FlexObjectTrait;
+    use FlexMediaTrait;
 
     /** @var Access|null */
     protected $_access;