Fix the styles for the code element