diff --git a/test/simple_table_mysql/test.sh b/test/simple_table_mysql/test.sh
index 9e2a1ab5..2e056cea 100755
--- a/test/simple_table_mysql/test.sh
+++ b/test/simple_table_mysql/test.sh
@@ -5,4 +5,4 @@ SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
 cd $SCRIPT_DIR
 
 cargo run --manifest-path ../../Cargo.toml -- \
--i schema.rs -o models -g id -g created_at -g updated_at -c "diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::prelude::MysqlConnection>>" --no-crud
+-i schema.rs -o models -g id -g created_at -g updated_at -c "diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::mysql::MysqlConnection>>" --no-crud
diff --git a/test/simple_table_pg/models/todos/generated.rs b/test/simple_table_pg/models/todos/generated.rs
index 166b38ab..2e777ec2 100644
--- a/test/simple_table_pg/models/todos/generated.rs
+++ b/test/simple_table_pg/models/todos/generated.rs
@@ -3,7 +3,7 @@
 use crate::diesel::*;
 use crate::schema::*;
 
-pub type ConnectionType = diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::prelude::PgConnection>>;
+pub type ConnectionType = diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::pg::PgConnection>>;
 
 /// Struct representing a row in table `todos`
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize, diesel::Queryable, diesel::Selectable, diesel::QueryableByName)]
diff --git a/test/simple_table_pg/test.sh b/test/simple_table_pg/test.sh
index bd818028..7af45024 100755
--- a/test/simple_table_pg/test.sh
+++ b/test/simple_table_pg/test.sh
@@ -5,4 +5,4 @@ SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
 cd $SCRIPT_DIR
 
 cargo run --manifest-path ../../Cargo.toml -- \
--i schema.rs -o models -g id -g created_at -g updated_at -c "diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::prelude::PgConnection>>"
+-i schema.rs -o models -g id -g created_at -g updated_at -c "diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::pg::PgConnection>>"
diff --git a/test/simple_table_sqlite/models/todos/generated.rs b/test/simple_table_sqlite/models/todos/generated.rs
index 1cd75bea..7d596a5f 100644
--- a/test/simple_table_sqlite/models/todos/generated.rs
+++ b/test/simple_table_sqlite/models/todos/generated.rs
@@ -3,7 +3,7 @@
 use crate::diesel::*;
 use crate::schema::*;
 
-pub type ConnectionType = diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::prelude::SqliteConnection>>;
+pub type ConnectionType = diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::sqlite::SqliteConnection>>;
 
 /// Struct representing a row in table `todos`
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize, diesel::Queryable, diesel::Selectable, diesel::QueryableByName)]
diff --git a/test/simple_table_sqlite/test.sh b/test/simple_table_sqlite/test.sh
index 43259a4e..0953e7d7 100755
--- a/test/simple_table_sqlite/test.sh
+++ b/test/simple_table_sqlite/test.sh
@@ -5,4 +5,4 @@ SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
 cd $SCRIPT_DIR
 
 cargo run --manifest-path ../../Cargo.toml -- \
--i schema.rs -o models -g id -g created_at -g updated_at -c "diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::prelude::SqliteConnection>>"
+-i schema.rs -o models -g id -g created_at -g updated_at -c "diesel::r2d2::PooledConnection<diesel::r2d2::ConnectionManager<diesel::sqlite::SqliteConnection>>"