Fix non-deterministic test

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