Fix flakey test when using yagnats

Signed-off-by: Phan Le <ple@pivotallabs.com>