Fixing mocha demo tests