Addition of the missing statement in mocha test