[FIX] Fix test suite