From 9f45182cfdbf64c9088e39a9d3dc02138093e252 Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Thu, 2 Feb 2023 12:55:32 +0100
Subject: [PATCH] fix tests

---
 router/src/db.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/router/src/db.rs b/router/src/db.rs
index 1c7f7399..34ded542 100644
--- a/router/src/db.rs
+++ b/router/src/db.rs
@@ -185,6 +185,7 @@ mod tests {
     use super::*;
     use text_generation_client::{NextTokenChooserParameters, StoppingCriteriaParameters};
     use tokio::sync::{mpsc, Semaphore};
+    use std::sync::Arc;
 
     fn default_entry() -> Entry {
         let semaphore = Arc::new(Semaphore::new(1));