AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAHEADmaster

2 files changed, 2 insertions, 1 deletions

diff --git a/src/main/java/space/m0e/quga/oop/lab56maven/decisions/GoToWorkstationDecision.java b/src/main/java/space/m0e/quga/oop/lab56maven/decisions/GoToWorkstationDecision.java
index 6dc4448..9002fc5 100644
--- a/src/main/java/space/m0e/quga/oop/lab56maven/decisions/GoToWorkstationDecision.java
+++ b/src/main/java/space/m0e/quga/oop/lab56maven/decisions/GoToWorkstationDecision.java
@@ -19,7 +19,6 @@ public class GoToWorkstationDecision extends Decision {
     public void cycle() {
         Workstation near = dwarf.findNearestWorkstation();
         if (near != null) {
-            System.out.println(near.getName());
             Platform.runLater(dwarf::updateItemLabel);
             dwarf.getFortress().removeFromContainer(dwarf);
             Coordinates nearCoordinates = dwarf.getCoordinatesToWorkstation(near);
diff --git a/src/main/java/space/m0e/quga/oop/lab56maven/decisions/GoTradingDecision.java b/src/main/java/space/m0e/quga/oop/lab56maven/decisions/GoTradingDecision.java
index 493fb45..784f061 100644
--- a/src/main/java/space/m0e/quga/oop/lab56maven/decisions/GoTradingDecision.java
+++ b/src/main/java/space/m0e/quga/oop/lab56maven/decisions/GoTradingDecision.java
@@ -33,10 +33,12 @@ public class GoTradingDecision extends Decision {
                 Item getItem = null;
                 switch (tradingItem) {
                     case WOOD -> {
+                        near.setWoodCount(near.getWoodCount() + amountOfItem);
                         near.setStoneCount(near.getStoneCount() - amountOfItem);
                         getItem = Item.STONE;
                     }
                     case STONE -> {
+                        near.setStoneCount(near.getStoneCount() + amountOfItem);
                         near.setWoodCount(near.getWoodCount() - amountOfItem);
                         getItem = Item.WOOD;
                     }