1 files changed, 3 insertions, 4 deletions
diff --git a/src/jrummikub/JRummikub.java b/src/jrummikub/JRummikub.java
index da2a0e9..c9047a1 100644
--- a/src/jrummikub/JRummikub.java
+++ b/src/jrummikub/JRummikub.java
@@ -1,7 +1,7 @@
 package jrummikub;
 
 import java.util.ArrayList;
-import java.util.Collections;
+import java.util.Arrays;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -65,7 +65,6 @@ public class JRummikub {
     stones.put(new Stone(0, StoneColor.BLACK, true), new Position(3.5f, 0));
     
     view.getPlayerPanel().getBoard().setStones(stones);
-    view.getPlayerPanel().getBoard().setHighlightedStones(Collections.singleton(stoneJoker));
     
     view.getPlayerPanel().getBoard().getClickEvent().add(new IListener1<Position>(){
 
@@ -99,8 +98,8 @@ public class JRummikub {
     stoneSets.put(new StoneSet(stoneList), new Position(3.5f, 4));
     
     view.getTable().setStoneSets(stoneSets);
-    view.getTable().setHighlightedStones(Collections.singleton(stone8));
-    
+
+    view.setSelectedStones(Arrays.asList(stoneJoker, stone8));
   }
 
 }