adding missing unit tests