fix typo
added some tests