CSS fix for code