Repair code box inside preview