Fixed unit tests after refactor to use grunt