fix to make it work again