fix race

this actually doesn't look like it's a race condition? the callback should be
one-to-one; it's not something registered that may trigger between runs...

Signed-off-by: Momchil Atanasov <m.momchil@gmail.com>