Update for Magento 2 beta