Update migration class.

nielssp · May 8, 2016 · d84fb5c · d84fb5c
1 parent 15846f0
commit d84fb5c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/Migration/Migration.php b/src/Migration/Migration.php
@@ -6,8 +6,8 @@
 namespace Jivoo\Data\Migration;
 
 use Jivoo\Data\Database\MigratableDatabase;
-use Jivoo\Data\Database\SchemaBuilder;
-use Jivoo\Models\DataType;
+use Jivoo\Data\DefinitionBuilder;
+use Jivoo\Data\DataType;
 
 /**
  * Base class for migrations.
@@ -71,14 +71,14 @@ public function __isset($table)
     /**
      * Create a table.
      *
-     * @param SchemaBuilder $schema
+     * @param DefinitionBuilder $definition
      *            Schema for table.
      */
-    protected function createTable(DefinitionBuilder $schema)
+    protected function createTable(DefinitionBuilder $definition)
     {
         try {
-            $this->db->createTable($schema);
-            $this->schema->createTable($schema);
+            $this->db->createTable($definition);
+            $this->schema->createTable($definition);
         } catch (\Exception $e) {
             if (! $this->ignoreExceptions) {
                 throw $e;