Fix broken example tests