1 files changed, 10 insertions, 5 deletions
diff --git a/server/src/db/private_message.rs b/server/src/db/private_message.rs
index b765bef4..e8b28485 100644
--- a/server/src/db/private_message.rs
+++ b/server/src/db/private_message.rs
@@ -1,6 +1,10 @@
-use super::*;
-use crate::apub::{make_apub_endpoint, EndpointType};
-use crate::schema::private_message;
+use crate::{
+  apub::{make_apub_endpoint, EndpointType},
+  db::Crud,
+  schema::private_message,
+};
+use diesel::{dsl::*, result::Error, *};
+use serde::{Deserialize, Serialize};
 
 #[derive(Queryable, Identifiable, PartialEq, Debug, Serialize, Deserialize)]
 #[table_name = "private_message"]
@@ -85,8 +89,9 @@ impl PrivateMessage {
 
 #[cfg(test)]
 mod tests {
-  use super::super::user::*;
-  use super::*;
+  use super::{super::user::*, *};
+  use crate::db::{establish_unpooled_connection, ListingType, SortType};
+
   #[test]
   fn test_crud() {
     let conn = establish_unpooled_connection();