Updated package to work again