diff --git a/tests/test_build_generation.py b/tests/test_build_generation.py
index 8fcc132..c207168 100644
--- a/tests/test_build_generation.py
+++ b/tests/test_build_generation.py
@@ -214,7 +214,7 @@ def test_file_permission_check_no_change(self):
             app_build_name = "my_app_2_1_1_2_1.tgz"
             assert os.path.isfile(app_build_name)
 
-            _, _, all_files, all_folders = self.extract_app_build(app_build_name)
+            _, _, all_files, _ = self.extract_app_build(app_build_name)
             assert "my_app_2/bin/file1.sh" in all_files
             assert get_file_permissions("my_app_2/bin/file1.sh") == "rwxr-xr-x"
             assert "my_app_2/bin/file2.sh" in all_files
@@ -232,7 +232,7 @@ def test_file_auto_change_permission(self):
             app_build_name = "my_app_2_1_1_2_1.tgz"
             assert os.path.isfile(app_build_name)
 
-            file_count, folder_count, all_files, all_folders = self.extract_app_build(app_build_name)
+            _, _, all_files, _ = self.extract_app_build(app_build_name)
             assert "my_app_2/bin/file1.sh" in all_files
             assert get_file_permissions("my_app_2/bin/file1.sh") == "rwxr-xr-x"
             assert "my_app_2/bin/file2.sh" in all_files
@@ -250,7 +250,7 @@ def test_file_permission_check_no_change_2(self):
             app_build_name = "my_app_2_1_1_2_1.tgz"
             assert os.path.isfile(app_build_name)
 
-            _, _, all_files, all_folders = self.extract_app_build(app_build_name)
+            _, _, all_files, _ = self.extract_app_build(app_build_name)
             assert "my_app_2/bin/file1.sh" in all_files
             assert get_file_permissions("my_app_2/bin/file1.sh") == "rwxr-xr-x"
             assert "my_app_2/bin/file2.sh" in all_files
@@ -268,7 +268,7 @@ def test_file_auto_change_permission_2(self):
             app_build_name = "my_app_2_1_1_2_1.tgz"
             assert os.path.isfile(app_build_name)
 
-            _, _, all_files, all_folders = self.extract_app_build(app_build_name)
+            _, _, all_files, _ = self.extract_app_build(app_build_name)
             assert "my_app_2/bin/file1.sh" in all_files
             assert get_file_permissions("my_app_2/bin/file1.sh") == "rwxr-xr-x"
             assert "my_app_2/bin/file2.sh" in all_files
@@ -294,7 +294,7 @@ def test_file_permission_change_via_user_commands(self):
             app_build_name = "my_app_2_1_1_2_1.tgz"
             assert os.path.isfile(app_build_name)
 
-            _, _, all_files, all_folders = self.extract_app_build(app_build_name)
+            _, _, all_files, _ = self.extract_app_build(app_build_name)
             assert "my_app_2/bin/file1.sh" in all_files
             assert get_file_permissions("my_app_2/bin/file1.sh") == "rwxr-xr-x"
             assert "my_app_2/bin/file2.sh" in all_files
@@ -320,7 +320,7 @@ def test_file_permission_change_via_user_commands_root_dir(self):
             app_build_name = "my_app_2_1_1_2_1.tgz"
             assert os.path.isfile(app_build_name)
 
-            _, _, all_files, all_folders = self.extract_app_build(app_build_name)
+            _, _, all_files, _ = self.extract_app_build(app_build_name)
             assert "my_app_2/bin/file1.sh" in all_files
             assert get_file_permissions("my_app_2/bin/file1.sh") == "rwxr-xr-x"
             assert "my_app_2/bin/file2.sh" in all_files