diff --git a/src/main/java/sk/ableneo/openslava2023/codefix/BalloonChallenge.java b/src/main/java/sk/ableneo/openslava2023/codefix/BalloonChallenge.java
index 9b3e0a3..979021f 100644
--- a/src/main/java/sk/ableneo/openslava2023/codefix/BalloonChallenge.java
+++ b/src/main/java/sk/ableneo/openslava2023/codefix/BalloonChallenge.java
@@ -18,8 +18,8 @@ public LinkedList findOrdersForProduct(Product p, boolean debug) {
             boolean found = false;
             if (order.getProducts().size() > 0) {
                 for (int j = 0; j <= order.getProducts().size(); j++) {
-                    Product p2 = (Product) order.getProducts().get(j);
-                    found = (p2 == p);
+                    Product p2 = order.getProducts().get(j);
+                    found = (p2.equals(p));
                 }
                 if (found && order != null)
                     l.add(order);