diff --git a/app/Http/Resources/DocumentResource.php b/app/Http/Resources/DocumentResource.php
index a4c085a..273c95f 100644
--- a/app/Http/Resources/DocumentResource.php
+++ b/app/Http/Resources/DocumentResource.php
@@ -16,28 +16,14 @@ class DocumentResource extends JsonResource
      */
     public function toArray($request)
     {
-        switch ($this->lang) {
-            case 1:
-                $tmpLang = "fr";
-                break;
-            case 2:
-                $tmpLang = "en";
-                break;
-            case 3:
-                $tmpLang = "de";
-                break;
-            case 4:
-                $tmpLang = "nl";
-                break;
-        }
         return [
             'id' => $this->id,
             'uuid' => $this->uuid,
             'title' => $this->title,
-            'type' => $this->type,
+            'type' => $this->type->value,
             'directory' => $this->directory_id,
             'path' => $this->path,
-            'lang' => $tmpLang,
+            'lang' => $this->lang->getDefinition(),
             'published' => ($this->status) ? 1 : 0,
             'parent' => $this->document_id,
             'created_at' => $this->created_at,