Fix the ugly white bg in code