Fix tests in package.json