Fix tests. Use mocha's assert instead of should.