diff --git a/code/TestCategory.php b/code/TestCategory.php index af3301b..a7726e6 100644 --- a/code/TestCategory.php +++ b/code/TestCategory.php @@ -15,7 +15,9 @@ class TestCategory extends DataObject { * Returns a dropdown map of all objects of this class */ static function map() { - return DataObject::get('TestCategory')->toDropdownMap(); + $categories = DataObject::get('TestCategory'); + if($categories) return $categories->toDropdownMap(); + else return array(); } function requireDefaultRecords(){