OpenWine task doesn't initialize correctly after the first demo

[adamconkey]

I was collecting multiple demos for the OpenWine task and noticed that it works fine on the first demo collected (cap is on bottle and robot removes cap from bottle), but on subsequent attempts the bottle is initialized with the cap on the table instead of the bottle, and the robot tries to plan its motion to the top of the bottle and then the environment just resets and puts the bottle in a different location. I've attached an image of the environment with the cap on the table, which is how it keeps being initialized after the first recorded demo.

[adamconkey]

Sorry I meant to put this on RLBench repo, will relocate to that.