Fixed stateful02 breakage.

Change-Id: Ifa1cc416dde2581c11b0c063ae03c57b3e6f1f63
Signed-off-by: Dana Kutenicsova <dkutenic@cisco.com>