fix flaky tests

Signed-off-by: Alex Jackson <ajackson@pivotallabs.com>