diff --git a/common/DiffractionExperiment.cpp b/common/DiffractionExperiment.cpp
index b523b27c..2096bf5d 100644
--- a/common/DiffractionExperiment.cpp
+++ b/common/DiffractionExperiment.cpp
@@ -330,20 +330,7 @@ DiffractionExperiment &DiffractionExperiment::MaskModuleEdges(bool input) {
 }
 
 DiffractionExperiment &DiffractionExperiment::Compression(JFJochProtoBuf::Compression input) {
-    switch (input) {
-        case JFJochProtoBuf::BSHUF_ZSTD:
-            dataset.set_compression(JFJochProtoBuf::BSHUF_ZSTD);
-            break;
-        case JFJochProtoBuf::BSHUF_ZSTD_RLE:
-            dataset.set_compression(JFJochProtoBuf::BSHUF_ZSTD_RLE);
-            break;
-        case JFJochProtoBuf::BSHUF_LZ4:
-            dataset.set_compression(JFJochProtoBuf::BSHUF_LZ4);
-            break;
-        default:
-            dataset.set_compression(JFJochProtoBuf::NO_COMPRESSION);
-            break;
-    }
+    dataset.set_compression(input);
     return *this;
 }