diff --git a/rival-evaluate/src/test/java/net/recommenders/rival/evaluation/metric/NDCGTest.java b/rival-evaluate/src/test/java/net/recommenders/rival/evaluation/metric/NDCGTest.java
index <HASH>..<HASH> 100644
--- a/rival-evaluate/src/test/java/net/recommenders/rival/evaluation/metric/NDCGTest.java
+++ b/rival-evaluate/src/test/java/net/recommenders/rival/evaluation/metric/NDCGTest.java
@@ -31,7 +31,7 @@ public class NDCGTest<U, V> {
                 predictions.addPreference((long) i, (long) j, (double) i * j % 5 + 1.0);
             }
         }
-        ndcg = new NDCG(predictions, test);
+        ndcg = new NDCG(predictions, test, new int[]{5, 10, 20});
     }
 
     @Test
@@ -41,6 +41,13 @@ public class NDCGTest<U, V> {
     }
 
     @Test
+    public void testComputeNDCGAt() {
+        ndcg.compute();
+        assertEquals(1.0, ndcg.getValue(5), 0.0);
+        assertEquals(1.0, ndcg.getValue(10), 0.0);
+    }
+
+    @Test
     public void testGetValuePerUser() {
         ndcg.compute();
         Map<Long, Double> ndcgPerUser = ndcg.getValuePerUser();
Fixed unit test for NDCG (see #<I>)