diff --git a/test/command/commandManager.test.ts b/test/command/commandManager.test.ts
index 92ac69d..d5a1f75 100644
--- a/test/command/commandManager.test.ts
+++ b/test/command/commandManager.test.ts
@@ -20,6 +20,7 @@ describe('CommandManager', () => {
       name: 'test',
       pattern: 'test',
       description: 'Test command',
+	  args: 0,
       handler: async (commandName: string, userInput: string) => {
         return 'Test result';
       },
@@ -34,6 +35,7 @@ describe('CommandManager', () => {
       name: 'test',
       pattern: 'test',
       description: 'Test command',
+	  args: 0,
       handler: async (commandName: string, userInput: string) => {
         return 'Test result';
       },
@@ -48,6 +50,7 @@ describe('CommandManager', () => {
       name: 'test',
       pattern: 'test',
       description: 'Test command',
+	  args: 0,
       handler: async (commandName: string, userInput: string) => {
         return 'Test result';
       },
@@ -64,6 +67,7 @@ describe('CommandManager', () => {
       pattern: 'customTest',
       description: 'Custom test command',
 	  message: 'Custom test result',
+	  args: 0,
       show: true,
 	  default: false,
 	  instructions: []
diff --git a/test/command/customCommand.test.ts b/test/command/customCommand.test.ts
index d013bef..e411c79 100644
--- a/test/command/customCommand.test.ts
+++ b/test/command/customCommand.test.ts
@@ -53,6 +53,7 @@ describe('CustomCommands', () => {
         description: 'Command 1',
         message: 'Command 1 message',
         default: false,
+		args: 0,
         show: true,
         instructions: ['instruction1', 'instruction2'],
       },
@@ -69,6 +70,7 @@ describe('CustomCommands', () => {
       description: 'Test command',
       message: 'Test message',
       default: false,
+	  args: 0,
       show: true,
       instructions: ['instruction1', 'instruction2'],
     };
@@ -84,6 +86,7 @@ describe('CustomCommands', () => {
       description: 'Test command',
       message: 'Test message',
       default: false,
+	  args: 0,
       show: true,
       instructions: ['instruction1', 'instruction2'],
     };
@@ -100,6 +103,7 @@ describe('CustomCommands', () => {
       description: 'Test command',
       message: 'Test message',
       default: false,
+	  args: 0,
       show: true,
       instructions: ['instruction1', 'instruction2'],
     };
@@ -116,6 +120,7 @@ describe('CustomCommands', () => {
       description: 'Test command',
       message: 'Test message "$1","$2"',
       default: false,
+	  args: 0,
       show: true,
       instructions: ['instruction1', 'instruction2'],
     };
diff --git a/test/handler/sendMessageBase.test.ts b/test/handler/sendMessageBase.test.ts
index fd449df..fd88d4f 100644
--- a/test/handler/sendMessageBase.test.ts
+++ b/test/handler/sendMessageBase.test.ts
@@ -114,7 +114,7 @@ describe('sendMessageBase', () => {
 			};
 
 			const result = await handlerResponseText(partialDataText, chatResponse);
-			expect(result).to.equal('Partial data\n\nError occurred!');
+			expect(result).to.equal('Error occurred!');
 		});
 	});