diff --git a/library/ZFDoctrine/Tool/DoctrineProvider.php b/library/ZFDoctrine/Tool/DoctrineProvider.php
index 1cf5e60..26e627e 100644
--- a/library/ZFDoctrine/Tool/DoctrineProvider.php
+++ b/library/ZFDoctrine/Tool/DoctrineProvider.php
@@ -279,10 +279,11 @@ protected function _createImportSchema()
 
     public function generateYamlFromModels()
     {
+        $doctrine = $this->_getDoctrineRegistry();
         $this->_loadDoctrineModels();
 
         $yamlDir = $this->_getYamlDirectoryPath();
-        Doctrine_Core::generateYamlFromModels($yamlDir);
+        Doctrine_Core::generateYamlFromModels($yamlDir, $doctrine->getModelPath());
 
         $this->_print('Successfully generated yaml schema files from model.', array('color' => 'green'));
         $this->_print('Destination Directory: ' . $yamlDir);