diff --git a/diesel/src/pg/types/json.rs b/diesel/src/pg/types/json.rs
index 059c86341ae6..c4246c58cdf6 100644
--- a/diesel/src/pg/types/json.rs
+++ b/diesel/src/pg/types/json.rs
@@ -46,9 +46,11 @@ impl ToSql<sql_types::Jsonb, Pg> for serde_json::Value {
     }
 }
 
-#[cfg(tests)]
+#[cfg(test)]
 mod tests {
-    use crate::query_builder::bind_types::ByteWrapper;
+    use super::*;
+    use crate::pg::{Pg, PgValue};
+    use crate::query_builder::bind_collector::ByteWrapper;
 
     #[test]
     fn json_to_sql() {