diff --git a/lib/registry.js b/lib/registry.js
index 6e0289e2..3cad9469 100644
--- a/lib/registry.js
+++ b/lib/registry.js
@@ -243,7 +243,7 @@ registry.createDataSource = function (name, options) {
     var BaseModel = settings.base || settings.super;
     if (!BaseModel) {
       // Check the connector types
-      var connectorTypes = ds.connector && ds.connector.getTypes();
+      var connectorTypes = ds.getTypes();
       if (Array.isArray(connectorTypes) && connectorTypes.indexOf('db') !== -1) {
         // Only set up the base model to PersistedModel if the connector is DB
         BaseModel = self.PersistedModel;