Fix tests for payment method