Action chunking fix so you can learn the proper goals