Fix typed nil
Typed nils are evil :) https://golang.org/doc/faq#nil_error
I knew I was using a typed nil here, but I was not realizing the wait group would cancel the context in that case too (of course!). Was just being too smart. Don't be too smart.
Edited by Mikhail Mazurskiy