Fixed broken example tests