fix couple of missing lines in test coverage