diff --git a/src/Concerns/WithPaymentProfile.php b/src/Concerns/WithPaymentProfile.php
index 6e871c1..d4ef7c4 100644
--- a/src/Concerns/WithPaymentProfile.php
+++ b/src/Concerns/WithPaymentProfile.php
@@ -33,7 +33,7 @@ public function updatePaymentProfile(Customer $customer, array $data = []): Paym
     /**
      * Deletes a customer payment profile from the payment gateway.
      */
-    public function deletePaymentProfile($customer, PaymentProfile $profile)
+    public function deletePaymentProfile(Customer $customer, PaymentProfile $profile)
     {
         throw new \LogicException('Method deletePaymentProfile must be implemented on your custom payment class.');
     }
diff --git a/src/Payments/Mollie.php b/src/Payments/Mollie.php
index b9c89ff..720e393 100644
--- a/src/Payments/Mollie.php
+++ b/src/Payments/Mollie.php
@@ -6,6 +6,8 @@
 use Igniter\Flame\Exception\ApplicationException;
 use Igniter\PayRegister\Classes\BasePaymentGateway;
 use Igniter\PayRegister\Concerns\WithPaymentRefund;
+use Igniter\PayRegister\Models\PaymentProfile;
+use Igniter\User\Models\Customer;
 use Illuminate\Support\Facades\Redirect;
 use Illuminate\Support\Facades\Response;
 use Mollie\Api\MollieApiClient;
@@ -209,7 +211,7 @@ public function processRefundForm($data, $order, $paymentLog)
     /**
      * {@inheritdoc}
      */
-    public function updatePaymentProfile($customer, $data)
+    public function updatePaymentProfile(Customer $customer, array $data = []): PaymentProfile
     {
         $profile = $this->model->findPaymentProfile($customer);
         $profileData = $profile ? (array)$profile->profile_data : [];
diff --git a/src/Payments/Square.php b/src/Payments/Square.php
index cc06e55..8b26ac4 100644
--- a/src/Payments/Square.php
+++ b/src/Payments/Square.php
@@ -155,7 +155,7 @@ public function processPaymentForm($data, $host, $order)
     /**
      * {@inheritdoc}
      */
-    public function supportsPaymentProfiles()
+    public function supportsPaymentProfiles(): bool
     {
         return true;
     }