fix unit test made flaky by recent refactoring

Signed-off-by: Johannes Petzold <jpetzold@pivotal.io>