Fix tests

Signed-off-by: Mike Gehard <mgehard@pivotallabs.com>