Fix missing test coverage

https://coveralls.io/files/187599888