Nice to see a multi-task version. If you give it the yellow-ball instruction on a scene set up for the red one, what does it do? Multi-task policies sometimes pick the task from the scene instead of the words, and that is easy to check with a few runs.
