diff --git a/receiver/jfjoch_action_test.cpp b/receiver/jfjoch_action_test.cpp
index cf020532..b1cd236a 100644
--- a/receiver/jfjoch_action_test.cpp
+++ b/receiver/jfjoch_action_test.cpp
@@ -28,16 +28,12 @@ int main(int argc, char **argv) {
     if (argc >= 5) nstreams = atoi(argv[4]);
     if (argc >= 6) processing_period = atoi(argv[5]);
 
-    std::vector<int64_t> detector_geom;
-    for (int i = 0; i < nstreams; i++)
-        detector_geom.push_back(nmodules);
-
-    DiffractionExperiment x(DetectorGeometry(nmodules, 2, 8, 36, true));
+    DiffractionExperiment x(DetectorGeometry(nmodules*nstreams, 2, 8, 36, true));
 
     x.Mode(DetectorMode::Conversion);
     x.ImagesPerTrigger(nimages).PedestalG0Frames(0).UseInternalPacketGenerator(true).PhotonEnergy_keV(12.4).NumTriggers(1);
     x.SpotFindingPeriod(std::chrono::milliseconds(processing_period)).MaskModuleEdges(false).MaskChipEdges(false);
-    x.Compression(JFJochProtoBuf::BSHUF_LZ4);
+    x.Compression(JFJochProtoBuf::BSHUF_LZ4).DataStreams(nstreams);
 
     std::vector<std::string> dev_name = {
             "/dev/jfjoch0",