Fix the rollback for parallel stream

Change-Id: I20432a9617581e1c3087a1989752e27aa226d071