[FIX] Tests