[FIX] Corrected tests