diff --git a/modules/media/edit.html.twig b/modules/media/edit.html.twig
index 9d14dd3..48a8730 100644
--- a/modules/media/edit.html.twig
+++ b/modules/media/edit.html.twig
@@ -1,17 +1,7 @@
 {% import '@SfsCms/macros/modules_edit.html.twig' as module %}
 
 <div {{ module.link_class(form.aligment_media, 'd-flex') }} {{ module.link_media_preview(form.media) }}>
-    {% if form.media.media.vars.data|default(false) and form.media.version.vars.data|default(false) %}
-        {% set versionSplit = form.media.version.vars.data|split('#') %}
-        {% set versionType = versionSplit[0]|default('image') %}
-        {% set versionKey = versionSplit[1]|default('_thumbnail') %}
-
-        {% if versionType == 'image' %}
-            {{ form.media.media.vars.data|sfs_media_render_image(versionKey, form.media.media.vars.attr) }}
-        {% elseif versionType == 'video' %}
-            {{ form.media.media.vars.data|sfs_media_render_video(versionKey, form.media.media.vars.attr) }}
-        {% elseif versionType == 'picture' %}
-            {{ form.media.media.vars.data|sfs_media_render_picture(versionKey, form.media.media.vars.attr) }}
-        {% endif %}
+    {% if form.media.media.vars.data|default(false) %}
+        {{ form.media.media.vars.data|sfs_media_render(form.media.version.vars.data, {'class':'img-fluid'}) }}
     {% endif %}
 </div>