diff --git a/lib/Doctrine/ORM/Tools/Cli/Tasks/SchemaToolTask.php b/lib/Doctrine/ORM/Tools/Cli/Tasks/SchemaToolTask.php
index 1cacfaaf2..1022c79dd 100644
--- a/lib/Doctrine/ORM/Tools/Cli/Tasks/SchemaToolTask.php
+++ b/lib/Doctrine/ORM/Tools/Cli/Tasks/SchemaToolTask.php
@@ -141,11 +141,11 @@ class SchemaToolTask extends AbstractTask
             );
         }
         
-        $metadataDriver = $this->getEntityManager()->getConfiguration()->getMetadataDriverImpl();
+        $metadataDriver = $em->getConfiguration()->getMetadataDriverImpl();
         
         if ($metadataDriver instanceof \Doctrine\ORM\Mapping\Driver\AnnotationDriver) {
-            if (isset($args['class-dir'])) {
-                $metadataDriver->setClassDirectory($args['class-dir']);
+            if (isset($arguments['class-dir'])) {
+                $metadataDriver->setClassDirectory($arguments['class-dir']);
             } else {
                 throw new CliException(
                     'The supplied configuration uses the annotation metadata driver. ' .