diff --git a/src/DependencyInjection/SetonoSyliusOrderEditExtension.php b/src/DependencyInjection/SetonoSyliusOrderEditExtension.php
index c26d265..fa1f30f 100644
--- a/src/DependencyInjection/SetonoSyliusOrderEditExtension.php
+++ b/src/DependencyInjection/SetonoSyliusOrderEditExtension.php
@@ -4,7 +4,6 @@
 
 namespace Setono\SyliusOrderEditPlugin\DependencyInjection;
 
-use Sylius\Bundle\CoreBundle\DependencyInjection\PrependDoctrineMigrationsTrait;
 use Symfony\Component\Config\FileLocator;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\DependencyInjection\Extension\Extension;
@@ -13,8 +12,6 @@
 
 final class SetonoSyliusOrderEditExtension extends Extension implements PrependExtensionInterface
 {
-    use PrependDoctrineMigrationsTrait;
-
     public function load(array $configs, ContainerBuilder $container): void
     {
         /**
@@ -34,7 +31,10 @@ public function prepend(ContainerBuilder $container): void
                     'after' => [
                         'setono_sylius_order_edit.set_initial_total' => [
                             'on' => ['create'],
-                            'do' => ['@setono_sylius_order_edit.order_processing.order_initial_total_processor', 'process'],
+                            'do' => [
+                                '@setono_sylius_order_edit.order_processing.order_initial_total_processor',
+                                'process',
+                            ],
                             'args' => ['object'],
                         ],
                     ],
@@ -53,22 +53,5 @@ public function prepend(ContainerBuilder $container): void
                 ],
             ],
         ]);
-
-        $this->prependDoctrineMigrations($container);
-    }
-
-    protected function getMigrationsNamespace(): string
-    {
-        return 'Setono\SyliusOrderEditPlugin\Migrations';
-    }
-
-    protected function getMigrationsDirectory(): string
-    {
-        return '@SetonoSyliusOrderEditPlugin/Migrations';
-    }
-
-    protected function getNamespacesOfMigrationsExecutedBefore(): array
-    {
-        return ['Sylius\Bundle\CoreBundle\Migrations'];
     }
 }
diff --git a/src/Entity/EditableOrderInterface.php b/src/Entity/EditableOrderInterface.php
new file mode 100644
index 0000000..9ca48ad
--- /dev/null
+++ b/src/Entity/EditableOrderInterface.php
@@ -0,0 +1,11 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Setono\SyliusOrderEditPlugin\Entity;
+
+use Sylius\Component\Core\Model\OrderInterface;
+
+interface EditableOrderInterface extends OrderInterface, InitialTotalAwareOrderInterface
+{
+}
diff --git a/src/Entity/InitialTotalAwareOrder.php b/src/Entity/InitialTotalAwareOrderTrait.php
similarity index 76%
rename from src/Entity/InitialTotalAwareOrder.php
rename to src/Entity/InitialTotalAwareOrderTrait.php
index 07c23b9..0b25f79 100644
--- a/src/Entity/InitialTotalAwareOrder.php
+++ b/src/Entity/InitialTotalAwareOrderTrait.php
@@ -6,9 +6,10 @@
 
 use Doctrine\ORM\Mapping as ORM;
 
-trait InitialTotalAwareOrder
+trait InitialTotalAwareOrderTrait
 {
-    /** @ORM\Column(type="integer", name="initial_total", options={"default": 0}) */
+    /** @ORM\Column(type="integer") */
+    #[ORM\Column(type: 'integer')]
     private int $initialTotal = 0;
 
     public function getInitialTotal(): int
diff --git a/src/Form/Type/OrderDiscountCollectionType.php b/src/Form/Type/OrderDiscountCollectionType.php
index 44ad775..0685509 100644
--- a/src/Form/Type/OrderDiscountCollectionType.php
+++ b/src/Form/Type/OrderDiscountCollectionType.php
@@ -4,6 +4,7 @@
 
 namespace Setono\SyliusOrderEditPlugin\Form\Type;
 
+use Setono\SyliusOrderEditPlugin\Model\OrderEditDiscountTypes;
 use Sylius\Component\Core\Model\OrderInterface;
 use Sylius\Component\Order\Factory\AdjustmentFactoryInterface;
 use Sylius\Component\Order\Model\AdjustmentInterface;
@@ -13,8 +14,6 @@
 
 final class OrderDiscountCollectionType extends AbstractType
 {
-    private const CUSTOM_ORDER_DISCOUNT = 'custom_order_discount';
-
     public function __construct(private readonly AdjustmentFactoryInterface $adjustmentFactory)
     {
     }
@@ -36,19 +35,19 @@ public function configureOptions(OptionsResolver $resolver): void
                 'label' => false,
             ],
             'getter' => function (OrderInterface &$order): array {
-                $adjustments = $order->getAdjustments(self::CUSTOM_ORDER_DISCOUNT)->toArray();
+                $adjustments = $order->getAdjustments(OrderEditDiscountTypes::SETONO_ADMIN_ORDER_DISCOUNT)->toArray();
 
                 return array_map(function (AdjustmentInterface $adjustment): int {
                     return -1 * $adjustment->getAmount();
                 }, $adjustments);
             },
             'setter' => function (OrderInterface &$order, array $discounts): void {
-                $order->removeAdjustments(self::CUSTOM_ORDER_DISCOUNT);
+                $order->removeAdjustments(OrderEditDiscountTypes::SETONO_ADMIN_ORDER_DISCOUNT);
 
                 /** @var int $discount */
                 foreach ($discounts as $discount) {
                     $adjustment = $this->adjustmentFactory->createWithData(
-                        self::CUSTOM_ORDER_DISCOUNT,
+                        OrderEditDiscountTypes::SETONO_ADMIN_ORDER_DISCOUNT,
                         'Custom discount: ' . number_format($discount, 2),
                         -1 * $discount,
                     );
diff --git a/src/Migrations/Version20240617131409.php b/src/Migrations/Version20240617131409.php
deleted file mode 100644
index a1148ed..0000000
--- a/src/Migrations/Version20240617131409.php
+++ /dev/null
@@ -1,31 +0,0 @@
-<?php
-
-declare(strict_types=1);
-
-namespace Setono\SyliusOrderEditPlugin\Migrations;
-
-use Doctrine\DBAL\Schema\Schema;
-use Doctrine\Migrations\AbstractMigration;
-
-/**
- * Auto-generated Migration: Please modify to your needs!
- */
-final class Version20240617131409 extends AbstractMigration
-{
-    public function getDescription(): string
-    {
-        return '';
-    }
-
-    public function up(Schema $schema): void
-    {
-        // this up() migration is auto-generated, please modify it to your needs
-        $this->addSql('ALTER TABLE sylius_order ADD initial_total INT DEFAULT 0 NOT NULL');
-    }
-
-    public function down(Schema $schema): void
-    {
-        // this down() migration is auto-generated, please modify it to your needs
-        $this->addSql('ALTER TABLE sylius_order DROP initial_total');
-    }
-}
diff --git a/src/Model/OrderEditDiscountTypes.php b/src/Model/OrderEditDiscountTypes.php
new file mode 100644
index 0000000..09075b1
--- /dev/null
+++ b/src/Model/OrderEditDiscountTypes.php
@@ -0,0 +1,14 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Setono\SyliusOrderEditPlugin\Model;
+
+final class OrderEditDiscountTypes
+{
+    public const SETONO_ADMIN_ORDER_DISCOUNT = 'setono_admin_order_discount';
+
+    private function __construct()
+    {
+    }
+}
diff --git a/src/OrderProcessing/OrderInitialTotalProcessor.php b/src/OrderProcessing/OrderInitialTotalProcessor.php
index 538d14a..dacd8d7 100644
--- a/src/OrderProcessing/OrderInitialTotalProcessor.php
+++ b/src/OrderProcessing/OrderInitialTotalProcessor.php
@@ -5,16 +5,15 @@
 namespace Setono\SyliusOrderEditPlugin\OrderProcessing;
 
 use Doctrine\ORM\EntityManagerInterface;
-use Setono\SyliusOrderEditPlugin\Entity\InitialTotalAwareOrderInterface;
-use Sylius\Component\Core\Model\OrderInterface;
+use Setono\SyliusOrderEditPlugin\Entity\EditableOrderInterface;
 
-final class OrderInitialTotalProcessor
+final class OrderInitialTotalProcessor implements OrderInitialTotalProcessorInterface
 {
     public function __construct(private readonly EntityManagerInterface $entityManager)
     {
     }
 
-    public function process(OrderInterface&InitialTotalAwareOrderInterface $order): void
+    public function process(EditableOrderInterface $order): void
     {
         $order->setInitialTotal($order->getTotal());
 
diff --git a/src/OrderProcessing/OrderInitialTotalProcessorInterface.php b/src/OrderProcessing/OrderInitialTotalProcessorInterface.php
new file mode 100644
index 0000000..85bc4ea
--- /dev/null
+++ b/src/OrderProcessing/OrderInitialTotalProcessorInterface.php
@@ -0,0 +1,12 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Setono\SyliusOrderEditPlugin\OrderProcessing;
+
+use Setono\SyliusOrderEditPlugin\Entity\EditableOrderInterface;
+
+interface OrderInitialTotalProcessorInterface
+{
+    public function process(EditableOrderInterface $order): void;
+}
diff --git a/tests/Application/config/packages/doctrine.yaml b/tests/Application/config/packages/doctrine.yaml
index 404acb1..fe21f8a 100644
--- a/tests/Application/config/packages/doctrine.yaml
+++ b/tests/Application/config/packages/doctrine.yaml
@@ -18,6 +18,6 @@ doctrine:
         mappings:
             App:
                 is_bundle: false
-                type: annotation
+                type: attribute
                 dir: '%kernel.project_dir%/src/Entity'
                 prefix: 'Setono\SyliusOrderEditPlugin\Tests\Application\Entity'
diff --git a/tests/Application/src/Entity/Order.php b/tests/Application/src/Entity/Order.php
index e922c4b..62493fd 100644
--- a/tests/Application/src/Entity/Order.php
+++ b/tests/Application/src/Entity/Order.php
@@ -5,15 +5,12 @@
 namespace Setono\SyliusOrderEditPlugin\Tests\Application\Entity;
 
 use Doctrine\ORM\Mapping as ORM;
-use Setono\SyliusOrderEditPlugin\Entity\InitialTotalAwareOrder;
-use Setono\SyliusOrderEditPlugin\Entity\InitialTotalAwareOrderInterface;
+use Setono\SyliusOrderEditPlugin\Entity\EditableOrderInterface;
+use Setono\SyliusOrderEditPlugin\Entity\InitialTotalAwareOrderTrait;
 
-/**
- * @ORM\Entity
- *
- * @ORM\Table(name="sylius_order")
- */
-class Order extends \Sylius\Component\Core\Model\Order implements InitialTotalAwareOrderInterface
+#[ORM\Entity]
+#[ORM\Table(name: 'sylius_order')]
+class Order extends \Sylius\Component\Core\Model\Order implements EditableOrderInterface
 {
-    use InitialTotalAwareOrder;
+    use InitialTotalAwareOrderTrait;
 }
diff --git a/tests/Unit/OrderProcessing/OrderInitialTotalProcessorTest.php b/tests/Unit/OrderProcessing/OrderInitialTotalProcessorTest.php
index 2983c1f..e0a5dc8 100644
--- a/tests/Unit/OrderProcessing/OrderInitialTotalProcessorTest.php
+++ b/tests/Unit/OrderProcessing/OrderInitialTotalProcessorTest.php
@@ -7,11 +7,8 @@
 use Doctrine\ORM\EntityManagerInterface;
 use PHPUnit\Framework\TestCase;
 use Prophecy\PhpUnit\ProphecyTrait;
-use Setono\SyliusOrderEditPlugin\Entity\InitialTotalAwareOrder;
-use Setono\SyliusOrderEditPlugin\Entity\InitialTotalAwareOrderInterface;
+use Setono\SyliusOrderEditPlugin\Entity\EditableOrderInterface;
 use Setono\SyliusOrderEditPlugin\OrderProcessing\OrderInitialTotalProcessor;
-use Sylius\Component\Core\Model\Order;
-use Sylius\Component\Core\Model\OrderInterface;
 
 final class OrderInitialTotalProcessorTest extends TestCase
 {
@@ -22,18 +19,13 @@ public function testItSetsInitialTotalAfterOrderIsCompleted(): void
         $entityManager = $this->prophesize(EntityManagerInterface::class);
 
         $processor = new OrderInitialTotalProcessor($entityManager->reveal());
-        $order = new class() extends Order implements OrderInterface, InitialTotalAwareOrderInterface {
-            use InitialTotalAwareOrder;
-
-            public function getTotal(): int
-            {
-                return 1000;
-            }
-        };
-        $entityManager->flush()->shouldBeCalled();
+        $order = $this->prophesize(EditableOrderInterface::class);
+
+        $order->getTotal()->willReturn(1000);
+        $order->setInitialTotal(1000)->shouldBeCalled();
 
-        $processor->process($order);
+        $entityManager->flush()->shouldBeCalled();
 
-        self::assertSame(1000, $order->getInitialTotal());
+        $processor->process($order->reveal());
     }
 }