Fixes formatting on training time.