diff --git a/engine/core/graphene/mutations.py b/engine/core/graphene/mutations.py
index 81bb917d..3078aaa0 100644
--- a/engine/core/graphene/mutations.py
+++ b/engine/core/graphene/mutations.py
@@ -228,9 +228,9 @@ class BuyOrder(BaseMutation):
             )
 
             match str(type(instance)):
-                case "<class 'payments.models.Transaction'>":
+                case "<class 'engine.payments.models.Transaction'>":
                     return BuyOrder(transaction=instance)
-                case "<class 'core.models.Order'>":
+                case "<class 'engine.core.models.Order'>":
                     return BuyOrder(order=instance)
                 case _:
                     raise TypeError(_(f"wrong type came from order.buy() method: {type(instance)!s}"))
@@ -493,9 +493,9 @@ class BuyWishlist(BaseMutation):
 
             instance = order.buy(force_balance=force_balance, force_payment=force_payment)
             match str(type(instance)):
-                case "<class 'payments.models.Transaction'>":
+                case "<class 'engine.payments.models.Transaction'>":
                     return BuyWishlist(transaction=instance)
-                case "<class 'core.models.Order'>":
+                case "<class 'engine.core.models.Order'>":
                     return BuyWishlist(order=instance)
                 case _:
                     raise TypeError(_(f"wrong type came from order.buy() method: {type(instance)!s}"))
@@ -535,9 +535,9 @@ class BuyProduct(BaseMutation):
         order.add_product(product_uuid=product_uuid, attributes=format_attributes(attributes))
         instance = order.buy(force_balance=force_balance, force_payment=force_payment)
         match str(type(instance)):
-            case "<class 'payments.models.Transaction'>":
+            case "<class 'engine.payments.models.Transaction'>":
                 return BuyProduct(transaction=instance)
-            case "<class 'core.models.Order'>":
+            case "<class 'engine.core.models.Order'>":
                 return BuyProduct(order=instance)
             case _:
                 raise TypeError(_(f"wrong type came from order.buy() method: {type(instance)!s}"))
diff --git a/engine/core/viewsets.py b/engine/core/viewsets.py
index bce34d89..11e25493 100644
--- a/engine/core/viewsets.py
+++ b/engine/core/viewsets.py
@@ -704,9 +704,9 @@ class OrderViewSet(EvibesViewSet):
                 chosen_products=serializer.validated_data.get("chosen_products"),
             )
             match str(type(instance)):
-                case "<class 'payments.models.Transaction'>":
+                case "<class 'engine.payments.models.Transaction'>":
                     return Response(status=status.HTTP_202_ACCEPTED, data=TransactionProcessSerializer(instance).data)
-                case "<class 'core.models.Order'>":
+                case "<class 'engine.core.models.Order'>":
                     return Response(status=status.HTTP_200_OK, data=OrderDetailSerializer(instance).data)
                 case _:
                     raise TypeError(_(f"wrong type came from order.buy() method: {type(instance)!s}"))