diff --git a/Models/CostCenter.php b/Models/CostCenter.php
index d357cb4..f36087d 100644
--- a/Models/CostCenter.php
+++ b/Models/CostCenter.php
@@ -159,7 +159,7 @@ class CostCenter
     /**
      * Set parent
      *
-     * @param string $parent Parent
+     * @param null|int|CostCenter $parent Parent
      *
      * @return void
      *
diff --git a/Models/CostObject.php b/Models/CostObject.php
index 1308f7b..82a3be9 100644
--- a/Models/CostObject.php
+++ b/Models/CostObject.php
@@ -159,7 +159,7 @@ class CostObject
     /**
      * Set parent
      *
-     * @param string $parent Parent
+     * @param null|int|CostObject $parent Parent
      *
      * @return void
      *