diff --git a/test/models/document.js b/test/models/document.js
index d5f1b0a..55d5f58 100644
--- a/test/models/document.js
+++ b/test/models/document.js
@@ -1,18 +1,11 @@
 /**
  * Sequelize model definition for testing User hasMany, Document belongsTo.
- */
-
-const Sequelize = require('sequelize');
-
-const { DataTypes } = Sequelize;
-
-/**
- * Initialize Profile definition
  *
  * @param sequelize Sequelize Instance
- * @returns {ProfileClass} Returns the Profile model
+ * @param Sequelize Sequelize Class
+ * @returns {DocumentClass} Returns the Document model
  */
-module.exports = sequelize => {
+module.exports = (sequelize, { DataTypes }) => {
   const Model = sequelize.define(
     'document',
     {
diff --git a/test/models/profile.js b/test/models/profile.js
index 68ecad2..6453fb9 100644
--- a/test/models/profile.js
+++ b/test/models/profile.js
@@ -1,18 +1,11 @@
 /**
  * Sequelize model definition for testing User hasOne, Profile belongsTo.
- */
-
-const Sequelize = require('sequelize');
-
-const { DataTypes } = Sequelize;
-
-/**
- * Initialize Profile definition
  *
  * @param sequelize Sequelize Instance
+ * @param Sequelize Sequelize Class
  * @returns {ProfileClass} Returns the Profile model
  */
-module.exports = sequelize => {
+module.exports = (sequelize, { DataTypes }) => {
   const Model = sequelize.define(
     'profile',
     {
diff --git a/test/models/user.js b/test/models/user.js
index 42c844d..50dd9af 100644
--- a/test/models/user.js
+++ b/test/models/user.js
@@ -1,3 +1,5 @@
+const supportedDataType = require('../utils/supported-datatype');
+
 /**
  * Sequelize attribute definitions for the `user` model.
  *
@@ -8,21 +10,12 @@
  * be present in the model when testing Sequelize v4.
  *
  * @see https://sequelize.org/master/manual/data-types.html
- */
-
-const Sequelize = require('sequelize');
-
-const { DataTypes } = Sequelize;
-
-const supportedDataType = require('../utils/supported-datatype');
-
-/**
- * Initialize User definition
  *
  * @param sequelize Sequelize Instance
+ * @param Sequelize Sequelize Class
  * @returns {UserClass} Returns the User model
  */
-module.exports = sequelize => {
+module.exports = (sequelize, { DataTypes }) => {
   const Model = sequelize.define(
     'user',
     {