diff --git a/src/algo.js b/src/algo.js
index bcc7e7b..dbb853c 100755
--- a/src/algo.js
+++ b/src/algo.js
@@ -82,7 +82,7 @@ function numberOfSets4(deck){
         for(j = i+1 ; j < deck.length - 2; j++){
             for(k = j+1 ; k < deck.length - 1 ; k++){
                 for(l = k + 1 ; l < deck.length;l++){
-                    if(isSet([deck[i],deck[j],deck[k]])){
+                    if(isSet([deck[i],deck[j],deck[k],deck[l]])){
                         //console.log(deck[i],deck[j],deck[k],deck[l])
                         res += 1
                     }
@@ -100,7 +100,7 @@ function numberOfSets5(deck){
             for(k = j+1 ; k < deck.length - 2 ; k++){
                 for(l = k + 1 ; l < deck.length - 1;l++){
                     for(m = l + 1; m <deck.length;m++){
-                        if(isSet([deck[i],deck[j],deck[k]])){
+                        if(isSet([deck[i],deck[j],deck[k],deck[l],deck[m]])){
                             //console.log(deck[i],deck[j],deck[k],deck[l],deck[m])
                             res += 1
                         }