diff --git a/src/main/java/models/Algo.java b/src/main/java/models/Algo.java
index 391cb4b..319ee10 100644
--- a/src/main/java/models/Algo.java
+++ b/src/main/java/models/Algo.java
@@ -63,11 +63,11 @@
 	public void inputSelectA(int guess, int attacker,  int target) {
 		this.succeedSelectA.inputSelectA(guess, attacker, target);
 	}
-	public void inputDrawA(int guess, List<Map.Entry<Object, Boolean>> b, int target) {
-		this.succeedDrawA.inputDrawA(guess, b, target);
+	public void inputDrawA(int guess, int target) {
+		this.succeedDrawA.inputDrawA(guess, target);
 	}
-	public void inputDrawB(int guess, List<Map.Entry<Object, Boolean>> a, int target) {
-		this.succeedDrawB.inputDrawB(guess, a, target);
+	public void inputDrawB(int guess, int target) {
+		this.succeedDrawB.inputDrawB(guess,  target);
 	}
 	public void inputSelectB(int guess, int attacker,  int target) {
 		this.succeedSelectB.inputSelectB(guess, attacker, target);
@@ -81,7 +81,7 @@
 	public Map.Entry<Boolean, Map.Entry<Integer, Integer>> getSucceedSelectA() {
 		return succeedSelectA.getValue();
 	}
-	public List getDeck() {
+	public List<Map.Entry<Integer, Boolean>> getDeck() {
 		return deck.getValue();
 	}
 	public Map.Entry<Boolean, Integer> getSucceedDrawB() {