diff --git a/tests/codex/proof/testdatasampler.nim b/tests/codex/proof/testdatasampler.nim
index 440de0ff..bf80af19 100644
--- a/tests/codex/proof/testdatasampler.nim
+++ b/tests/codex/proof/testdatasampler.nim
@@ -169,8 +169,11 @@ asyncchecksuite "Test proof datasampler - components":
 
 asyncchecksuite "Test proof datasampler - main":
   let
-    numberOfSlotBlocks = 4 # 16
-    totalNumberOfSlots = 2 #4
+    # The number of slot blocks and number of slots, combined with
+    # the bytes per block, make it so that there are exactly 256 cells
+    # in the dataset.
+    numberOfSlotBlocks = 4
+    totalNumberOfSlots = 2
     datasetSlotIndex = 1
     localStore = CacheStore.new()
 
diff --git a/tests/codex/proof/testdatasampler_expected.nim b/tests/codex/proof/testdatasampler_expected.nim
index 71000600..189dbb72 100644
--- a/tests/codex/proof/testdatasampler_expected.nim
+++ b/tests/codex/proof/testdatasampler_expected.nim
@@ -19,5 +19,5 @@ proc getExpectedBlockToCellProofs*(): seq[string] =
 
 proc getExpectedSampleData*(): string =
   "B2".repeat(CellSize) &
-  "59".repeat(CellSize) &
+  "95".repeat(CellSize) &
   "EE".repeat(CellSize)