Added missing test coverage, #28