Test coverage fixed