diff --git a/Controller/ApiController.php b/Controller/ApiController.php
index 6030957..c608815 100755
--- a/Controller/ApiController.php
+++ b/Controller/ApiController.php
@@ -18,6 +18,7 @@ use Modules\Admin\Models\AccountMapper;
 use Modules\Admin\Models\NullAccount;
 use Modules\Media\Models\CollectionMapper;
 use Modules\Media\Models\MediaMapper;
+use Modules\Media\Models\NullCollection;
 use Modules\Media\Models\NullMedia;
 use Modules\Media\Models\Reference;
 use Modules\Media\Models\ReferenceMapper;
@@ -190,7 +191,7 @@ final class ApiController extends Controller
                 if ($collection === null) {
                     $collection = MediaMapper::getParentCollection($path)->limit(1)->execute();
 
-                    if ($collection instanceof NullMedia) {
+                    if ($collection instanceof NullCollection) {
                         $collection = $this->app->moduleManager->get('Media')->createRecursiveMediaCollection(
                             '/Modules/Media/Files',
                             $accountPath,
@@ -220,7 +221,7 @@ final class ApiController extends Controller
                 if ($collection === null) {
                     $collection = MediaMapper::getParentCollection($path)->limit(1)->execute();
 
-                    if ($collection instanceof NullMedia) {
+                    if ($collection instanceof NullCollection) {
                         $collection = $this->app->moduleManager->get('Media')->createRecursiveMediaCollection(
                             '/Modules/Media/Files',
                             $path,