Fix js unit test