diff --git a/src/tests/gitlab/ckpt1/NGramToNextChoicesMapTests.java b/src/tests/gitlab/ckpt1/NGramToNextChoicesMapTests.java
index 24b65af22e1229aa957fab3ea433475c32bcd622..cc9657960184475778206e24fb088ab752a104ce 100644
--- a/src/tests/gitlab/ckpt1/NGramToNextChoicesMapTests.java
+++ b/src/tests/gitlab/ckpt1/NGramToNextChoicesMapTests.java
@@ -95,7 +95,7 @@ public class NGramToNextChoicesMapTests extends TestsUtility {
             if (items.length != answer.length) return 0;
             String[] itemsWithoutCounts = new String[items.length];
             for (int j = 0; j < answer.length; j++) {
-                if (items[j].value != 1) return 0;
+                if (!items[j].value.equals(1)) return 0;
                 itemsWithoutCounts[j] = items[j].key;
             }
             Arrays.sort(itemsWithoutCounts);
@@ -200,7 +200,7 @@ public class NGramToNextChoicesMapTests extends TestsUtility {
                     return 0;
                 }
                 // checks if correct count for given word after
-                if (expected[j].value != results[j].value) {
+                if (!expected[j].value.equals(results[j].value)) {
                     return 0;
                 }
             }