Fix code display