add two minor fixes (#243)

* add two minor fixes

* bump

Author: tenub

components/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@schematichq/schematic-components",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "main": "dist/schematic-components.cjs.js",
   "module": "dist/schematic-components.esm.js",
   "types": "dist/schematic-components.d.ts",

components/src/components/shared/checkout-dialog/Checkout.tsx

@@ -105,7 +105,12 @@ export const Checkout = ({
             <Text $size={18}>{t("Add payment method")}</Text>
           </Box>
 
-          <PaymentForm onConfirm={(value) => setPaymentMethodId(value)} />
+          <PaymentForm
+            onConfirm={(value) => {
+              setPaymentMethodId(value);
+              togglePaymentForm();
+            }}
+          />
 
           {data.subscription?.paymentMethod && (
             <Box>

components/src/components/shared/payment-form/PaymentForm.tsx

@@ -84,6 +84,7 @@ export const PaymentForm = ({ onConfirm }: PaymentFormProps) => {
         id="submit"
         disabled={isLoading || !stripe || !elements || isConfirmed}
         isLoading={isLoading}
+        style={{ flexShrink: 0 }}
         $color="primary"
       >
         {isLoading ? t("Loading") : t("Save payment method")}