diff --git a/src/main/controllers/web/drive/directory.tmpl.html b/src/main/controllers/web/drive/directory.tmpl.html
index 86d1784..54f2136 100644
--- a/src/main/controllers/web/drive/directory.tmpl.html
+++ b/src/main/controllers/web/drive/directory.tmpl.html
@@ -3,15 +3,26 @@
{#for .dir.children.? |$child| =}
{#switch $child case dir |$dir| =}
+ |
|
{$dir.name.?} |
- {#case file |$file| =}
+ {#case file |$file|}
+
+ {= #if %user |$u|}
+ {#if $u.avatar_file_id == $file.meta.id =}
+
+ {= #elif $u.header_file_id == $file.meta.id =}
+
+ {= /if =}
+ {= /if =}
+ |
{= #if $file.meta.sensitive =}
- |
+
{= #else =}
-
+
{= /if =}
+
{$file.name.?} |
{#if $file.meta.content_type |$t|}{$t}{/if} |
{$file.meta.size} |