were you able to solve this issue? I am getting quite a similar problem but following the training. Cheers!