fix trainbase