Fix the lost the code styling