[misc] Update tests