diff --git a/worklenz-frontend/src/components/task-list-v2/TaskListV2Table.tsx b/worklenz-frontend/src/components/task-list-v2/TaskListV2Table.tsx
index 99fb5bb1..172f1b74 100644
--- a/worklenz-frontend/src/components/task-list-v2/TaskListV2Table.tsx
+++ b/worklenz-frontend/src/components/task-list-v2/TaskListV2Table.tsx
@@ -515,7 +515,7 @@ const TaskListV2Section: React.FC = () => {
             return (
               <div
                 key={column.id}
-                className={`text-sm font-semibold text-gray-600 dark:text-gray-300 ${
+                className={`text-sm font-semibold text-gray-600 dark:text-gray-300 border-r border-gray-200 dark:border-gray-700 ${
                   column.id === 'dragHandle'
                     ? 'flex items-center justify-center'
                     : column.id === 'checkbox'
@@ -549,7 +549,7 @@ const TaskListV2Section: React.FC = () => {
           })}
           {/* Add Custom Column Button - positioned at the end and scrolls with content */}
           <div
-            className="flex items-center justify-center px-2"
+            className="flex items-center justify-center px-2 border-r border-gray-200 dark:border-gray-700"
             style={{ width: '50px', flexShrink: 0 }}
           >
             <AddCustomColumnButton />
diff --git a/worklenz-frontend/src/hooks/useTaskSocketHandlers.ts b/worklenz-frontend/src/hooks/useTaskSocketHandlers.ts
index cc55829b..c00fa14d 100644
--- a/worklenz-frontend/src/hooks/useTaskSocketHandlers.ts
+++ b/worklenz-frontend/src/hooks/useTaskSocketHandlers.ts
@@ -232,7 +232,7 @@ export const useTaskSocketHandlers = () => {
         dispatch(
           updateTask({
             ...currentTask,
-            status: newStatusValue,
+            status: response.status_id || newStatusValue, // Use actual status_id instead of category
             progress: response.complete_ratio || currentTask.progress,
             updatedAt: new Date().toISOString(),
           })
@@ -806,13 +806,7 @@ export const useTaskSocketHandlers = () => {
           task_key: data.task_key || '',
           title: data.name || '',
           description: data.description || '',
-          status: (data.status_category?.is_todo
-            ? 'todo'
-            : data.status_category?.is_doing
-              ? 'doing'
-              : data.status_category?.is_done
-                ? 'done'
-                : 'todo') as 'todo' | 'doing' | 'done',
+          status: data.status || 'todo',
           priority: (data.priority_value === 3
             ? 'critical'
             : data.priority_value === 2