diff --git a/public/site/config/routes/upload-pdf.php b/public/site/config/routes/upload-pdf.php
index bfd5143..ff4b9a2 100644
--- a/public/site/config/routes/upload-pdf.php
+++ b/public/site/config/routes/upload-pdf.php
@@ -46,7 +46,7 @@ return [
         "pdf" => $newFile->id()
       ]);
 
-      return $newPage->toArray();
+      return getFileData($newFile);
 
     } catch (Exception $e) {
       return [
diff --git a/src/components/comments/Comment.vue b/src/components/comments/Comment.vue
index 4158934..a62960f 100644
--- a/src/components/comments/Comment.vue
+++ b/src/components/comments/Comment.vue
@@ -30,7 +30,7 @@
         }}
       </p>
       <div
-        v-if="comment.author.uuid === userStore.user.uuid"
+        v-if="userStore.canEditComment(comment)"
         class="comment__ctas | mt-8"
       >
         <button class="btn btn--transparent btn--icon btn--sm" data-icon="edit">
diff --git a/src/stores/brief.js b/src/stores/brief.js
index 5f2e97c..e256c22 100644
--- a/src/stores/brief.js
+++ b/src/stores/brief.js
@@ -1,11 +1,10 @@
-import { defineStore } from "pinia";
-import { ref } from "vue";
+import { defineStore, storeToRefs } from "pinia";
 import { usePageStore } from "./page";
 
 export const useBriefStore = defineStore("brief", () => {
   async function addPdf(event, briefUri, redirectToParent = false) {
     const file = event.target.files[0];
-    const { page } = usePageStore();
+    const { page } = storeToRefs(usePageStore());
 
     if (file) {
       const formData = new FormData();
@@ -20,7 +19,7 @@ export const useBriefStore = defineStore("brief", () => {
         const result = await response.json();
 
         if (response.ok) {
-          page.value = result;
+          page.value.steps[0].files.push(result);
         } else {
           console.error("Error uploading file:", result.error);
         }
diff --git a/src/stores/user.js b/src/stores/user.js
index 2577942..524ce95 100644
--- a/src/stores/user.js
+++ b/src/stores/user.js
@@ -24,5 +24,15 @@ export const useUserStore = defineStore("user", () => {
     });
   }
 
-  return { user, notifications, readNotification, readAllNotifications };
+  function canEditComment(comment) {
+    return user.value.uuid === comment.author.uuid;
+  }
+
+  return {
+    user,
+    notifications,
+    readNotification,
+    readAllNotifications,
+    canEditComment,
+  };
 });
diff --git a/src/views/Kanban.vue b/src/views/Kanban.vue
index 49def36..f94b243 100644
--- a/src/views/Kanban.vue
+++ b/src/views/Kanban.vue
@@ -16,7 +16,7 @@ import Header from "../components/project/Header.vue";
 import DialogWrapper from "../components/project/DialogWrapper.vue";
 import { usePageStore } from "../stores/page";
 import { storeToRefs } from "pinia";
-import { computed, watch } from "vue";
+import { watch } from "vue";
 import { useDialogStore } from "../stores/dialog";
 import { useRoute } from "vue-router";