task is working fine for both flows