fix typo
fix tests