fix(test): Added unit tests

Added the missing unit tests.

fixes #2