diff --git a/main.py b/main.py
index 2c0231e..4fbc252 100644
--- a/main.py
+++ b/main.py
@@ -156,16 +156,17 @@ def run_egm(para: Parameter, animation=None) -> dict:
             rated_voltage = para.rated_voltage
             logger.info(f"交、直流标识{para.ac_or_dc}")
             for u_bar in range(voltage_n):  # 计算不同工作电压下的跳闸率
-                # if para.ac_or_dc=="AC":
-                #     # TODO 需要区分交、直流
-                #     u_ph = (
-                #         math.sqrt(2)
-                #         * rated_voltage
-                #         * math.cos(2 * math.pi / voltage_n * u_bar)
-                #         / 1.732
-                #     )  # 运行相电压
-                # else:
-                u_ph = rated_voltage/1.732
+                if para.ac_or_dc=="AC":
+                    # TODO 需要区分交、直流
+                    # u_ph = (
+                    #     math.sqrt(2)
+                    #     * rated_voltage
+                    #     * math.cos(2 * math.pi / voltage_n * u_bar)
+                    #     / 1.732
+                    # )  # 运行相电压
+                    u_ph = rated_voltage/1.732
+                else:
+                    u_ph = rated_voltage
                 logger.info(f"计算第{phase_conductor_foo + 1}相，电压为{u_ph:.2f}kV")
                 # 迭代法计算最大电流
                 i_max = 0