Fixing tests

Accindentally pushed previous commit

Signed-off-by: Maria Shaldibina <mariash@pivotallabs.com>