diff --git a/src/main/java/org/avmframework/examples/inputdatageneration/triangle/TriangleBranchTargetObjectiveFunction.java b/src/main/java/org/avmframework/examples/inputdatageneration/triangle/TriangleBranchTargetObjectiveFunction.java
index f5862f7..9d6ec8a 100644
--- a/src/main/java/org/avmframework/examples/inputdatageneration/triangle/TriangleBranchTargetObjectiveFunction.java
+++ b/src/main/java/org/avmframework/examples/inputdatageneration/triangle/TriangleBranchTargetObjectiveFunction.java
@@ -62,7 +62,7 @@ private Triangle.TriangleType classify(int num1, int num2, int num3) {
       num2 = temp;
     }
 
-    if (trace.greaterThan(2, num1, num2)) {
+    if (trace.greaterThan(2, num1, num3)) {
       int temp = num1;
       num1 = num3;
       num3 = temp;