Magento: In cart page small image instead of item thumbnail image

<img src="<?php echo $this->helper('catalog/image')->init(Mage::getModel('catalog/product')->load($_item->getProductId()), 'small_image')->resize(336); ?>" alt="<?php echo $this->escapeHtml($this->getProductName()) ?>" />