Fix issue with calling get_shipping_method