Fixup integration tests after YT change

Change-Id: Ib1daa9617c4f10e08f82674ed64a557d4813998b
Signed-off-by: Robert Varga <rovarga@cisco.com>