diff --git a/src/Elcodi/Store/CartBundle/Controller/CheckoutController.php b/src/Elcodi/Store/CartBundle/Controller/CheckoutController.php
index 8aa49e78..0762e022 100644
--- a/src/Elcodi/Store/CartBundle/Controller/CheckoutController.php
+++ b/src/Elcodi/Store/CartBundle/Controller/CheckoutController.php
@@ -259,7 +259,7 @@ public function paymentAction(CartInterface $cart)
          */
         $paymentMethods = $this
             ->get('elcodi.wrapper.payment_methods')
-            ->get();
+            ->get($cart);
 
         /**
          * Available shipping methods
@@ -306,7 +306,8 @@ public function paymentAction(CartInterface $cart)
     /**
      * Checkout shipping range
      *
-     * @param CartInterface $cart Cart
+     * @param CartInterface $cart           Cart
+     * @param string        $shippingMethod Shipping method
      *
      * @return Response Response
      *
@@ -324,8 +325,10 @@ public function paymentAction(CartInterface $cart)
      *      name = "cart"
      * )
      */
-    public function applyShippingMethodAction(CartInterface $cart, $shippingMethod)
-    {
+    public function applyShippingMethodAction(
+        CartInterface $cart,
+        $shippingMethod
+    ) {
         /**
          * Desired shipping method
          */