TST: Improve test coverage

Add coverage for some paths that are missed in base