diff --git a/src/main/java/cz/cvut/fel/sit/omo/foodchain/product/Product.java b/src/main/java/cz/cvut/fel/sit/omo/foodchain/product/Product.java
index 74dc178..3d21dd0 100644
--- a/src/main/java/cz/cvut/fel/sit/omo/foodchain/product/Product.java
+++ b/src/main/java/cz/cvut/fel/sit/omo/foodchain/product/Product.java
@@ -1,12 +1,13 @@
 package cz.cvut.fel.sit.omo.foodchain.product;
 
 import cz.cvut.fel.sit.omo.foodchain.operation.IOperation;
+import cz.cvut.fel.sit.omo.foodchain.operation.OperationBase;
 
 public class Product
 {
     private ProductState productState;
     private ProductType productType;
-    private IOperation currentOperation;
+    private OperationBase currentOperation;
 
     private String name;
 
@@ -112,12 +113,12 @@ public class Product
         this.name = name;
     }
 
-    public IOperation getCurrentOperation()
+    public OperationBase getCurrentOperation()
     {
         return currentOperation;
     }
 
-    public void setCurrentOperation(IOperation currentOperation)
+    public void setCurrentOperation(OperationBase currentOperation)
     {
         this.currentOperation = currentOperation;
     }