Fix missing test coverage