Fixed typo
Added small test