fix test
fix typo in code