[FIX] Correcting tests.