[FIX] Make testing work again