diff --git a/src/Gateway/GatewayModule.php b/src/Gateway/GatewayModule.php
index d73dfb51..e06c151e 100644
--- a/src/Gateway/GatewayModule.php
+++ b/src/Gateway/GatewayModule.php
@@ -6,7 +6,6 @@
 
 namespace Mollie\WooCommerce\Gateway;
 
-use Automattic\WooCommerce\Admin\Overrides\Order;
 use Automattic\WooCommerce\Internal\DataStores\Orders\CustomOrdersTableController;
 use Mollie\WooCommerce\Vendor\Inpsyde\Modularity\Module\ExecutableModule;
 use Mollie\WooCommerce\Vendor\Inpsyde\Modularity\Module\ModuleClassNameIdTrait;
@@ -639,9 +638,9 @@ public function in3FieldsMandatory($fields, $errors)
     }
 
     /**
-     * @param Order $order
+     * @param $order
      */
-    public function in3FieldsMandatoryPayForOrder(Order $order)
+    public function in3FieldsMandatoryPayForOrder($order)
     {
         $paymentMethod = filter_input(INPUT_POST, 'payment_method', FILTER_SANITIZE_SPECIAL_CHARS) ?? false;