diff --git a/project/src/neural/infrence.rs b/project/src/neural/infrence.rs
index 9c7477f..e5b8601 100644
--- a/project/src/neural/infrence.rs
+++ b/project/src/neural/infrence.rs
@@ -1,4 +1,4 @@
-use crate::{data::MnistBatcher, model::Model, training::TrainingConfig};
+use super::{data::MnistBatcher, model::Model, training::TrainingConfig};
 use burn::{
     data::{dataloader::batcher::Batcher, dataset::vision::MnistItem},
     prelude::*,
diff --git a/project/src/neural/training.rs b/project/src/neural/training.rs
index 682e9bb..b9b350c 100644
--- a/project/src/neural/training.rs
+++ b/project/src/neural/training.rs
@@ -1,4 +1,4 @@
-use crate::{
+use super::{
     data::{MnistBatch, MnistBatcher},
     model::{Model, ModelConfig},
 };