bug fix: train chapter was not cleaning up after itself