Mocha test change

Look good @humphd?