1 files changed, 6 insertions, 5 deletions
diff --git a/backend/app/src/utils/calculate_category_balance.rs b/backend/app/src/utils/calculate_category_balance.rs
index 3c1c62f..e1de57e 100644
--- a/backend/app/src/utils/calculate_category_balance.rs
+++ b/backend/app/src/utils/calculate_category_balance.rs
@@ -1,4 +1,7 @@
-use crate::models::{budget_update::BudgetUpdate, category::Category};
+use budgeting_app_core::models::{
+    budget_update::BudgetUpdate,
+    category::Category
+};
 
 use super::{calculate_budgets_accrual, find_by_id_or};
 
@@ -19,12 +22,10 @@ pub fn calculate_category_balance(
 
 #[cfg(test)]
 mod test {
+    use budgeting_app_core::models::{budget_update::BudgetUpdate, category::Category, date_time_utc::DateTimeUtc};
     use chrono::{Days, Utc};
 
-    use crate::{
-        models::{budget_update::BudgetUpdate, category::Category, date_time_utc::DateTimeUtc},
-        utils::calculate_category_balance,
-    };
+    use crate::utils::calculate_category_balance;
 
     #[test]
     fn when_no_data_then_returns_zero() {