fix typo and update tests.