diff --git a/src/components/Badge/AssignTableRow.jsx b/src/components/Badge/AssignTableRow.jsx
index 626c3ac4d6..fe20bf8f92 100644
--- a/src/components/Badge/AssignTableRow.jsx
+++ b/src/components/Badge/AssignTableRow.jsx
@@ -27,7 +27,6 @@ function AssignTableRow(props) {
     setSelect(prev => (prev !== next ? next : prev));
   }, [effectiveSelected, props.badge?._id]);
 
-
   const badgeId = props.badge?._id;
   const domId = `assign-badge-${badgeId}`;
 
@@ -41,7 +40,8 @@ function AssignTableRow(props) {
   }, []);
 
   const isSelected = selectedBadges.includes(badgeId);
-  console.log(selectedBadges,"sele",badgeId,props)
+  // eslint-disable-next-line no-console
+  console.log(selectedBadges, 'sele', badgeId, props);
 
   const handleCheckBoxChange = e => {
     if (e.target.checked) {
diff --git a/src/components/Badge/__tests__/AssignBadge.test.jsx b/src/components/Badge/__tests__/AssignBadge.test.jsx
index d3d86f89e2..43c2dabcbe 100644
--- a/src/components/Badge/__tests__/AssignBadge.test.jsx
+++ b/src/components/Badge/__tests__/AssignBadge.test.jsx
@@ -1,7 +1,7 @@
 import React from 'react';
 import { render, screen, fireEvent, waitFor } from '@testing-library/react';
 import thunk from 'redux-thunk';
-import configureStore from 'redux-mock-store';
+import { configureStore } from 'redux-mock-store';
 import { Provider } from 'react-redux';
 import { vi } from 'vitest';
 
@@ -107,10 +107,8 @@ describe('AssignBadge component', () => {
     const input = screen.getByPlaceholderText('Full Name');
     fireEvent.change(input, { target: { value: 'Jerry' } });
 
-    await waitFor(() => {
-      const userRows = screen.getAllByRole('row').slice(1); // Exclude header row
-      fireEvent.click(userRows[0]);
-    });
+    const userRows = (await screen.findAllByRole('row')).slice(1); // Exclude header row
+    fireEvent.click(userRows[0]);
 
     const alertText = screen.getByRole('alert').textContent;
     expect(alertText).toContain('1 user(s) selected');
@@ -126,10 +124,8 @@ describe('AssignBadge component', () => {
     const input = screen.getByPlaceholderText('Full Name');
     fireEvent.change(input, { target: { value: 'Jerry' } });
 
-    await waitFor(() => {
-      const userRows = screen.getAllByRole('row').slice(1); // Exclude header row
-      fireEvent.click(userRows[0]);
-    });
+    const userRows = (await screen.findAllByRole('row')).slice(1); // Exclude header row
+    fireEvent.click(userRows[0]);
 
     expect(screen.getByText('Assign Badge')).toBeEnabled();
   });
diff --git a/src/components/Projects/WBS/SingleTask/SingleTask.jsx b/src/components/Projects/WBS/SingleTask/SingleTask.jsx
index 2ef50e25f9..91d8e69556 100644
--- a/src/components/Projects/WBS/SingleTask/SingleTask.jsx
+++ b/src/components/Projects/WBS/SingleTask/SingleTask.jsx
@@ -151,7 +151,7 @@ function SingleTask(props) {
               <tbody className={darkMode ? 'bg-yinmn-blue' : ''}>
                 <tr>
                   <th scope="row">
-                    <div className="d-flex flex-column">
+                    <div className="d-flex flex-column align-items-start">
                       <EditTaskModal
                         key={`editTask_${task._id}`}
                         parentNum={task.num}
@@ -169,9 +169,8 @@ function SingleTask(props) {
                           <Button
                             type="button"
                             size="sm"
-                            className="btn btn-danger"
+                            className="btn btn-danger mt-1 ml-2" 
                             onClick={() => showUpDeleteModal()}
-                            style={darkMode ? boxStyleDark : boxStyle}
                           >
                             Delete
                           </Button>