Fix tests...

Test case: Ran `npm test`. No errors.