diff --git a/modules/ppcp-save-payment-methods/src/SavePaymentMethodsModule.php b/modules/ppcp-save-payment-methods/src/SavePaymentMethodsModule.php
index abdea60fa..5137a2b68 100644
--- a/modules/ppcp-save-payment-methods/src/SavePaymentMethodsModule.php
+++ b/modules/ppcp-save-payment-methods/src/SavePaymentMethodsModule.php
@@ -62,13 +62,16 @@ class SavePaymentMethodsModule implements ModuleInterface {
 		$billing_agreements_endpoint = $c->get( 'api.endpoint.billing-agreements' );
 		assert( $billing_agreements_endpoint instanceof BillingAgreementsEndpoint );
 
-		add_action( 'woocommerce_paypal_payments_gateway_migrate_on_update', function() use($settings, $billing_agreements_endpoint) {
-			$reference_transaction_enabled = $billing_agreements_endpoint->reference_transaction_enabled();
-			if ( $reference_transaction_enabled !== true ) {
-				$settings->set( 'vault_enabled', false );
-				$settings->persist();
+		add_action(
+			'woocommerce_paypal_payments_gateway_migrate_on_update',
+			function() use ( $settings, $billing_agreements_endpoint ) {
+				$reference_transaction_enabled = $billing_agreements_endpoint->reference_transaction_enabled();
+				if ( $reference_transaction_enabled !== true ) {
+					$settings->set( 'vault_enabled', false );
+					$settings->persist();
+				}
 			}
-		});
+		);
 
 		if (
 			( ! $settings->has( 'vault_enabled' ) || ! $settings->get( 'vault_enabled' ) )