Cover missing lines in tests