diff --git a/main.py b/main.py
index 2de9164..38ac496 100644
--- a/main.py
+++ b/main.py
@@ -140,14 +140,16 @@ def egm():
             logger.debug(f"最低相防护标识{rg_type}")
             rated_voltage = para.rated_voltage
             for u_bar in range(voltage_n):  # 计算不同工作电压下的跳闸率
-                # 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
+                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
                 logger.info(f"计算第{phase_conductor_foo + 1}相，电压为{u_ph:.2f}kV")
                 # 迭代法计算最大电流
                 i_max = 0