1 files changed, 3 insertions, 3 deletions
diff --git a/src/solvers/gpu/mod.rs b/src/solvers/gpu/mod.rs
index f147f31..2c7f69d 100644
--- a/src/solvers/gpu/mod.rs
+++ b/src/solvers/gpu/mod.rs
@@ -4,7 +4,7 @@ pub mod output;
 
 pub use manager::*;
 
-type MaskMessage = (u64, Vec<u64>);
+type MaskMessage = (u64, u32, Vec<u64>);
 type RowMessage = (u64, Vec<Vec<u32>>);
 
 pub enum Message {
@@ -33,7 +33,7 @@ impl ResultMessage {
             id,
         }
     }
-    fn valid_walls(&self) -> &[Vec<u32>] {
+    fn valid_walls(&self) -> Vec<Vec<u32>> {
         let mut result = vec![Vec::new(); self.wg_size];
         for (j, r) in self.data.iter().enumerate() {
             for b in 0..64 {
@@ -44,7 +44,7 @@ impl ResultMessage {
                 }
             }
         }
-        result.as_ref()
+        result
     }
 }