Fix css classes