[FIX] fixed broken tests