diff --git a/scripts/DXAI/objectives.cs b/scripts/DXAI/objectives.cs
index 93814d3..04229fb 100644
--- a/scripts/DXAI/objectives.cs
+++ b/scripts/DXAI/objectives.cs
@@ -380,10 +380,10 @@ function AIEnhancedFlagCaptureTask::monitor(%task, %client)
       return;
     
     %client.isMovingToTarget = true;
-   // if (%client.targetCaptureFlag.getObjectMount() != %client.player)
-    //    %client.setMoveTarget(%client.targetCaptureFlag.getPosition());
-   //// else
-       %client.setMoveTarget(nameToID("Team" @ %client.team @ "Flag1").getPosition());
+    if (%client.targetCaptureFlag.getObjectMount() != %client.player)
+        %client.setMoveTarget(%client.targetCaptureFlag.getPosition());
+    else
+        %client.setMoveTarget(nameToID("Team" @ %client.team @ "Flag1").getPosition());
 }
 //------------------------------------------------------------------------------------------