diff --git a/services/SchematicService.php b/services/SchematicService.php
index 04d5eea2..c18f3153 100644
--- a/services/SchematicService.php
+++ b/services/SchematicService.php
@@ -49,12 +49,12 @@ public function exportToYaml($file)
     /**
      * Import data model.
      *
-     * @param array $model
-     * @param bool  $force if set to true items not in the import will be deleted
+     * @param Schematic_DataModel $model
+     * @param bool                $force if set to true items not in the import will be deleted
      *
      * @return Schematic_ResultModel
      */
-    private function importDataModel(array $model, $force)
+    private function importDataModel(Schematic_DataModel $model, $force)
     {
         // Import schema
         $pluginImportResult = craft()->schematic_plugins->import($model->plugins);