Add assertion to flaky spec

Signed-off-by: Chris Wang <cwang@pivotallabs.com>