diff --git a/src/compute/models/linreg.rs b/src/compute/models/linreg.rs
index 5add9f9..ba76197 100644
--- a/src/compute/models/linreg.rs
+++ b/src/compute/models/linreg.rs
@@ -34,10 +34,10 @@ impl LinReg {
     }
 }
 
+#[cfg(test)]
 mod tests {
 
-    use super::LinReg;
-    use crate::matrix::{Matrix};
+    use super::*;
 
     #[test]
     fn test_linreg_fit_predict() {