fix coverage and one broken test