fix coverage, include missing lines