Updates tests to work with "npm test"