diff --git a/src/main/java/controls/AbstractGameState.java b/src/main/java/controls/AbstractGameState.java
index cd6b9d5..1332c29 100644
--- a/src/main/java/controls/AbstractGameState.java
+++ b/src/main/java/controls/AbstractGameState.java
@@ -1,6 +1,7 @@
 package controls;
 
 import resources.Algo;
+import values.Card;
 
 import java.util.List;
 import java.util.Map;
@@ -20,15 +21,15 @@
     public abstract void attack();
 
 
-    public List<Map.Entry<Integer, Boolean>> getMyHands() {
+    public List<Card> getMyHands() {
         return algo.getHandsA();
     }
 
-    public List<Map.Entry<Integer, Boolean>> getOpponentHands() {
+    public List<Card> getOpponentHands() {
         return algo.getHandsB();
     }
 
-    public Map.Entry<Integer, Boolean> getTopCard() {
+    public Card getTopCard() {
         if (isDeckLess()) return null;
         return algo.getDeck().get(0);
     }