Fix for SingleSubFlow test.