diff --git a/Views/Notes/createNote.phtml b/Views/Notes/createNote.phtml
index db595b0..5b47333 100644
--- a/Views/Notes/createNote.phtml
+++ b/Views/Notes/createNote.phtml
@@ -8,7 +8,7 @@ $isEditMode = false;
 $note = null;
 if ($isEditMode) {
     $noteId = $_GET['id'] ?? 0;
-    $note = $this->notesModel->getNoteById($noteId, 2); //$_SESSION['user_id']
+    $note = $this->notesModel->getNoteById($noteId, $_SESSION['user_id']);
     if (!$note) {
         echo "<div class='alert alert-danger'>Note not found or you don't have permission to edit it.</div>";
         echo "<a href='?controller=Notes&page=showNotes&do=showNotes' class='button secondary'>Back to Dashboard</a>";
diff --git a/Views/Notes/editNote.phtml b/Views/Notes/editNote.phtml
index 3108b9e..b990bd1 100644
--- a/Views/Notes/editNote.phtml
+++ b/Views/Notes/editNote.phtml
@@ -8,7 +8,7 @@ $isEditMode = true;
 $note = null;
 if ($isEditMode) {
     $noteId = $_GET['id'] ?? 0;
-    $note = $this->notesModel->getNoteById($noteId, 2); //$_SESSION['user_id']
+    $note = $this->notesModel->getNoteById($noteId, $_SESSION['user_id']);
     if (!$note) {
         echo "<div class='alert alert-danger'>Note not found or you don't have permission to edit it.</div>";
         echo "<a href='?controller=Notes&page=showNotes&do=showNotes' class='button secondary'>Back to Dashboard</a>";