Improve code coverage to meet the quality gate

[kploch]

^{Created from JetBrains using CodeStream}