diff --git a/tests/unit_tests/test_main.py b/tests/unit_tests/test_main.py
index ece668b618..bfecd03ea4 100644
--- a/tests/unit_tests/test_main.py
+++ b/tests/unit_tests/test_main.py
@@ -1657,7 +1657,11 @@ def test_upload_program_ota_partition_table_with_file_arg(
     assert exit_code == 0
     assert host == "192.168.1.100"
     mock_run_ota.assert_called_once_with(
-        ["192.168.1.100"], 3232, None, Path("partitions.bin"), OTA_TYPE_UPDATE_PARTITION_TABLE
+        ["192.168.1.100"],
+        3232,
+        None,
+        Path("partitions.bin"),
+        OTA_TYPE_UPDATE_PARTITION_TABLE,
     )
 
 
@@ -3240,7 +3244,11 @@ def test_upload_program_ota_static_ip_with_mqttip(
         tmp_path / ".esphome" / "build" / "test" / ".pioenvs" / "test" / "firmware.bin"
     )
     mock_run_ota.assert_called_once_with(
-        ["192.168.1.100", "192.168.2.50"], 3232, None, expected_firmware, OTA_TYPE_UPDATE_APP
+        ["192.168.1.100", "192.168.2.50"],
+        3232,
+        None,
+        expected_firmware,
+        OTA_TYPE_UPDATE_APP,
     )