Corrected typo build_all_train