diff --git a/patches/WEB-2301---get-repository-of-API-models.diff b/patches/WEB-2301---get-repository-of-API-models.diff
index 31c2ee34f..a47a44327 100644
--- a/patches/WEB-2301---get-repository-of-API-models.diff
+++ b/patches/WEB-2301---get-repository-of-API-models.diff
@@ -1,13 +1,13 @@
 diff --git a/src/Repositories/ModuleRepository.php b/src/Repositories/ModuleRepository.php
-index cb6986e0..bf268cc9 100644
+index 47e56466..5172e25d 100644
 --- a/src/Repositories/ModuleRepository.php
 +++ b/src/Repositories/ModuleRepository.php
-@@ -867,6 +867,8 @@ abstract class ModuleRepository
-         if (!$modelOrRepository) {
-             if (class_exists($relation) && (new $relation) instanceof Model) {
-                 $modelOrRepository = str_after_last($relation, '\\');
-+            } elseif (class_exists($relation) && (new $relation) instanceof \App\Libraries\Api\Models\BaseApiModel) {
-+                $modelOrRepository = 'Api\\' .str_after_last($relation, '\\');
+@@ -696,6 +696,8 @@ abstract class ModuleRepository
+         if (! $modelOrRepository) {
+             if (class_exists($relation) && (new $relation()) instanceof Model) {
+                 $modelOrRepository = Str::afterLast($relation, '\\');
++            } elseif (class_exists($relation) && (new $relation()) instanceof \App\Libraries\Api\Models\BaseApiModel) {
++                $modelOrRepository = 'Api\\' . Str::afterLast($relation, '\\');
              } else {
                  $morphedModel = Relation::getMorphedModel($relation);
-                 if (class_exists($morphedModel) && (new $morphedModel) instanceof Model) {
+                 if (class_exists($morphedModel) && (new $morphedModel()) instanceof Model) {