Training samples should run ok with the new framework structure