Fixing example tests