diff --git a/src/main/java/ruscz/game/gameOfProd/Game.java b/src/main/java/ruscz/game/gameOfProd/Game.java
index e54b3d1..1c83253 100644
--- a/src/main/java/ruscz/game/gameOfProd/Game.java
+++ b/src/main/java/ruscz/game/gameOfProd/Game.java
@@ -51,7 +51,7 @@ public class Game extends JFrame {
 //            battle.startBattle();
 
         //Tohle spustí na začátku tileset overview
-//            tilesetOverview = new TilesetOverview();
+            tilesetOverview = new TilesetOverview();
 
         setDefaultCloseOperation(EXIT_ON_CLOSE);
         setUndecorated(true);
diff --git a/src/main/java/ruscz/game/gameOfProd/LevelEditor/TilesetOverview.java b/src/main/java/ruscz/game/gameOfProd/LevelEditor/TilesetOverview.java
index 19b43aa..84ff5e5 100644
--- a/src/main/java/ruscz/game/gameOfProd/LevelEditor/TilesetOverview.java
+++ b/src/main/java/ruscz/game/gameOfProd/LevelEditor/TilesetOverview.java
@@ -140,7 +140,7 @@ public class TilesetOverview extends JFrame {
             tileset.setBounds(0, tilesetOffset, width + (Xtiles * 6), height + (Ytiles * 6));
             tilesetOffset += height + + (Ytiles * 6);
             tilesetPanel.add(tileset);
-            int panelWidth = tilesetPanel.getWidth();
+            int panelWidth = tilesetPanel.getPreferredSize().width;
             if (panelWidth < width + (Xtiles * 6)) panelWidth = width + (Xtiles * 6);
             tilesetPanel.setPreferredSize(new Dimension(panelWidth, tilesetOffset));
         }