modified test to pass in mocha