Add minor improvement in UT

Run mocha test in the start method
fix solution