diff --git a/src/dataset/mod.rs b/src/dataset/mod.rs
index da790b4..31a12cf 100644
--- a/src/dataset/mod.rs
+++ b/src/dataset/mod.rs
@@ -83,7 +83,7 @@ pub(crate) fn deserialize_data(
 ) -> Result<(Vec<f32>, Vec<f32>, usize, usize), io::Error> {
     // read the same file back into a Vec of bytes
     let (num_samples, num_features) = {
-        let mut buffer = [0u8; 8];
+        let mut buffer = [0u8; if cfg!(target_arch = "wasm32") { 4 } else { 8 }];
         buffer.copy_from_slice(&bytes[0..8]);
         let num_features = usize::from_le_bytes(buffer);
         buffer.copy_from_slice(&bytes[8..16]);