diff --git a/tests/test_main.py b/tests/test_main.py
index b572557..48d9ea8 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -21,9 +21,11 @@ class TestCreateChat(unittest.TestCase):
         self.tags = ['test_tag']
 
     @patch('os.listdir')
+    @patch('pathlib.Path.iterdir')
     @patch('builtins.open')
-    def test_create_chat_with_tags(self, open_mock, listdir_mock):
+    def test_create_chat_with_tags(self, open_mock, iterdir_mock, listdir_mock):
         listdir_mock.return_value = ['testfile.txt']
+        iterdir_mock.return_value = [pathlib.Path(x) for x in listdir_mock.return_value]
         open_mock.return_value.__enter__.return_value = io.StringIO(dump_data(
             {'question': 'test_content', 'answer': 'some answer',
              'tags': ['test_tag']}))
@@ -41,9 +43,11 @@ class TestCreateChat(unittest.TestCase):
                          {'role': 'user', 'content': self.question})
 
     @patch('os.listdir')
+    @patch('pathlib.Path.iterdir')
     @patch('builtins.open')
-    def test_create_chat_with_other_tags(self, open_mock, listdir_mock):
+    def test_create_chat_with_other_tags(self, open_mock, iterdir_mock, listdir_mock):
         listdir_mock.return_value = ['testfile.txt']
+        iterdir_mock.return_value = [pathlib.Path(x) for x in listdir_mock.return_value]
         open_mock.return_value.__enter__.return_value = io.StringIO(dump_data(
             {'question': 'test_content', 'answer': 'some answer',
              'tags': ['other_tag']}))
@@ -57,9 +61,11 @@ class TestCreateChat(unittest.TestCase):
                          {'role': 'user', 'content': self.question})
 
     @patch('os.listdir')
+    @patch('pathlib.Path.iterdir')
     @patch('builtins.open')
-    def test_create_chat_without_tags(self, open_mock, listdir_mock):
+    def test_create_chat_without_tags(self, open_mock, iterdir_mock, listdir_mock):
         listdir_mock.return_value = ['testfile.txt', 'testfile2.txt']
+        iterdir_mock.return_value = [pathlib.Path(x) for x in listdir_mock.return_value]
         open_mock.side_effect = (
             io.StringIO(dump_data({'question': 'test_content',
                                    'answer': 'some answer',