This fix is also needed around line 74. Can you do that? Also, the cumprod op also need it. While you are at it, can you fix it too?

thanks