Fixing a typo
Adding unit tests for 100% coverage